a

app.py

@@ -35,7 +35,7 @@ def generate(paper_text):
     ).to(model.device)
 
     generated_ids = model.generate(
-        **model_inputs,
+        input_ids=model_inputs,
         max_new_tokens=256
     )
     generated_ids = [