Update app.py

app.py

@@ -65,7 +65,7 @@ def generate_answer(question, context, ground_truth):
 
     # Generate the answer
     with torch.no_grad():
-        generated_ids = model.generate(input_ids, max_new_tokens=max_target_length)
+        generated_ids = model.generate(input_ids=input_ids, max_new_tokens=max_target_length)
 
     # Decode and return the generated answer
     generated_answer = tokenizer.decode(generated_ids[0], skip_special_tokens=True)