Update app.py

app.py

@@ -17,8 +17,8 @@ proc_tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-4k-instruct
 
 proc_pipe = pipeline(
     "text-generation",
-    model=model,
-    tokenizer=tokenizer,
+    model=proc_model,
+    tokenizer=proc_tokenizer,
 )
 
 generation_args = {
@@ -28,12 +28,16 @@ generation_args = {
     "do_sample": False,
 }
 
+def generate_response(inputs):
+    output = proc_pipe(inputs, **generation_args)
+    return output[0]['generated_text']
+
 
 
 # Create a Gradio interface
 iface = gr.Interface(
-    fn=proc_pipe, 
-    inputs=[gr.Textbox(lines=5, placeholder="Enter your prompt here..."), **generation_args]
+    fn=generate_response, 
+    inputs=[gr.Textbox(lines=5, placeholder="Enter your prompt here...")]
     outputs=gr.Textbox()
 )