Dofla commited on
Commit
5d6fa12
·
verified ·
1 Parent(s): 6ed4fcd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -14,6 +14,7 @@ def answer_question(context, question):
14
  start_logits = outputs.start_logits
15
  end_logits = outputs.end_logits
16
 
 
17
  # Assurez-vous que les logits sont des tenseurs
18
  start_index = torch.argmax(start_logits, dim=1).item()
19
  end_index = torch.argmax(end_logits, dim=1).item() + 1
 
14
  start_logits = outputs.start_logits
15
  end_logits = outputs.end_logits
16
 
17
+
18
  # Assurez-vous que les logits sont des tenseurs
19
  start_index = torch.argmax(start_logits, dim=1).item()
20
  end_index = torch.argmax(end_logits, dim=1).item() + 1