Update app.py

app.py

@@ -17,7 +17,7 @@ model = PeftModel.from_pretrained(model, peft_model_id)
 
 
 def make_inference(abstract):
-  batch = tokenizer(f"Keywords:\n", return_tensors='pt')
+  batch = tokenizer(f"Below is an abstract\n\n### Abstract:\n{abstract}\n### Keywords:\n", return_tensors='pt')
 
   with torch.cuda.amp.autocast():
     output_tokens = model.generate(**batch, max_new_tokens=50)