Update app.py

app.py

@@ -25,8 +25,7 @@ model = PeftModel.from_pretrained(base_model, model_repo)
 # Load tokenizer from the Hub repo
 tokenizer = AutoTokenizer.from_pretrained(model_repo, use_fast=False)
 
-def generate(prompt, max_tokens=128, temperature=0.7, top_p=0.9):
-    # try:
+def generate_response(prompt, max_tokens=128, temperature=0.7, top_p=0.9):
     inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
     outputs = model.generate(
         **inputs,
@@ -36,8 +35,6 @@ def generate(prompt, max_tokens=128, temperature=0.7, top_p=0.9):
         do_sample=True
     )
     return tokenizer.decode(outputs[0], skip_special_tokens=True)
-    # except Exception as e:
-    #     return f"Error: {str(e)}"
 
 iface = gr.Interface(
     fn=generate_response,