Update app.py

app.py

@@ -196,13 +196,20 @@ def create_context(dataframe, filters):
 
 
 # Функция для получения ответа от модели
-def get_model_response(question, context):
+def get_model_response(question, context=""):
     inputs = tokenizer.encode_plus(question, context, return_tensors="pt", max_length=512, truncation=True)
     outputs = model(**inputs)
     answer_start = torch.argmax(outputs.start_logits)
     answer_end = torch.argmax(outputs.end_logits) + 1
+
+    # Если модель не нашла ответа, возвращаем сообщение об ошибке
+    if answer_start >= answer_end:
+        return "Не удалось найти подходящий объект."
+
     answer = tokenizer.convert_tokens_to_string(tokenizer.convert_ids_to_tokens(inputs["input_ids"][0][answer_start:answer_end]))
-    return answer
+    return answer.strip() if answer.strip() != "[CLS]" else "Не найдено."
+
+    
 
 
 # Главная функция для Gradio