Update app.py
Browse files
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}
|