Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|