mindspark121 commited on
Commit
a7ec6ba
·
verified ·
1 Parent(s): 2518806

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -79,4 +79,4 @@ def summarize_chat(request: ChatRequest):
79
  inputs = summarization_tokenizer("summarize: " + request.message, return_tensors="pt", max_length=4096, truncation=True)
80
  summary_ids = summarization_model.generate(inputs.input_ids, max_length=500, num_beams=4, early_stopping=True)
81
  summary = summarization_tokenizer.decode(summary_ids[0], skip_special_tokens=True)
82
- return {"summary": summary}
 
79
  inputs = summarization_tokenizer("summarize: " + request.message, return_tensors="pt", max_length=4096, truncation=True)
80
  summary_ids = summarization_model.generate(inputs.input_ids, max_length=500, num_beams=4, early_stopping=True)
81
  summary = summarization_tokenizer.decode(summary_ids[0], skip_special_tokens=True)
82
+ return {"summary": summary}