Update app.py

app.py

@@ -66,7 +66,7 @@ def generate_answer(question, context):
 
     # Generate the answer
     with torch.no_grad():
-        generated_ids = model.generate(input_ids)
+        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)