Update app.py

app.py

@@ -87,13 +87,16 @@ def process_query(query_text: str, vectorstore):
         # Updated HuggingFaceEndpoint configuration
         # Используем модель t5-base вместо google/flan-t5-base
         model = HuggingFaceEndpoint(
-            repo_id="t5-base",  # Модель, поддерживающая text2text-generation
+            repo_id="deepset/roberta-base-squad2",  # Модель для question-answering
             task="question-answering",
             temperature=0.5,
             max_length=512,
-        #    huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
+            # huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
         )
-        response_text = model.invoke(prompt)  # Use invoke instead of predict
+
+# Передаем вопрос и контекст в виде словаря
+response = model.invoke({"question": query_text, "context": context_text})
+response_text = response.get("answer", "Нет ответа")
         
         sources = list(set([doc.metadata.get("source", "") for doc, _ in results]))
         return response_text, sources