Update app.py

app.py

@@ -25,7 +25,7 @@ model = PeftModel.from_pretrained(model, model_id)
 def greet(text):
     with torch.no_grad():
         # Include EOS token for better context
-        input_text = "<s>### User:\n{text}\n\n### Assistant:\n"
+        input_text = f"<s>### User:\n{text}\n\n### Assistant:\n"
         
         batch = tokenizer(input_text, return_tensors='pt', add_special_tokens=True).to(device)
 
@@ -44,5 +44,6 @@ def greet(text):
         response_parts = response.split("### Assistant:")
         return response_parts[0]  # Return only the first part
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text", title="PEFT Model for Big Brain")
+iface = gr.Interface(fn=greet, inputs="text", outputs="text"
+, title="PEFT Model for Big Brain")
 iface.launch()  # Share directly to Gradio Space