Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -91,7 +91,7 @@ def summarize_text(results):
|
|
| 91 |
if summarizer is None:
|
| 92 |
return "Summarization model could not be loaded."
|
| 93 |
|
| 94 |
-
summary = summarizer('
|
| 95 |
return summary
|
| 96 |
except Exception as e:
|
| 97 |
logger.error(f"Summarization failed: {str(e)}")
|
|
@@ -109,7 +109,7 @@ def answer_question(context, question):
|
|
| 109 |
return "Please enter your Question"
|
| 110 |
messages = [
|
| 111 |
{"role": "system", "content": "You are a helpful assistant who can answer questions based on the given context."},
|
| 112 |
-
{"role": "user", "content": f"Context: {'
|
| 113 |
]
|
| 114 |
|
| 115 |
response = qa_pipeline(messages, max_new_tokens=256)[0]['generated_text']
|
|
|
|
| 91 |
if summarizer is None:
|
| 92 |
return "Summarization model could not be loaded."
|
| 93 |
|
| 94 |
+
summary = summarizer(''.join(d['translated'] for d in results if 'translated' in d), max_length=150, min_length=50, do_sample=False)[0]['summary_text']
|
| 95 |
return summary
|
| 96 |
except Exception as e:
|
| 97 |
logger.error(f"Summarization failed: {str(e)}")
|
|
|
|
| 109 |
return "Please enter your Question"
|
| 110 |
messages = [
|
| 111 |
{"role": "system", "content": "You are a helpful assistant who can answer questions based on the given context."},
|
| 112 |
+
{"role": "user", "content": f"Context: {''.join(d['translated'] for d in context if 'translated' in d)}\n\nQuestion: {question}"}
|
| 113 |
]
|
| 114 |
|
| 115 |
response = qa_pipeline(messages, max_new_tokens=256)[0]['generated_text']
|