Update app.py

app.py

@@ -21,12 +21,10 @@ def qa_pipeline(text, question):
     }
 
     # Выполнение предсказания
-    with torch.no_grad():
-        outputs = model(**batch)
-    
-    # Извлечение логитов начала и конца ответа
-    start_logits = outputs.start_logits
-    end_logits = outputs.end_logits
+    start_logits, end_logits, loss = model(batch)
+
+    start_index = torch.argmax(start_logits, dim=-1).item()
+    end_index = torch.argmax(end_logits, dim=-1).item()
     
     # Нахождение индексов начала и конца ответа
     start_index = torch.argmax(start_logits, dim=-1).item()