Update app.py

app.py

@@ -10,6 +10,9 @@ model = AutoModelForQuestionAnswering.from_pretrained("Dofla/distilbert-squad")
 def answer_question(context, question):
     inputs = tokenizer.encode_plus(question, context, return_tensors="pt", padding=True, truncation=True)
     start_logits, end_logits = model(**inputs)
+    outputs = model(**inputs)
+    start_logits = outputs.start_logits
+    end_logits = outputs.end_logits
 
     # Assurez-vous que les logits sont des tenseurs
     start_index = torch.argmax(start_logits, dim=1).item()