a

app.py

@@ -30,7 +30,8 @@ def generate(paper_text):
     ]
     model_inputs = tokenizer.apply_chat_template(
         messages,
-        add_generation_prompt=True
+        add_generation_prompt=True,
+        return_tensors='pt'
     ).to(model.device)
 
     generated_ids = model.generate(