Update app.py

app.py

@@ -5,25 +5,23 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 model = AutoModelForCausalLM.from_pretrained(
     "microsoft/Phi-3-mini-4k-instruct", 
     trust_remote_code=True  # Allow custom code execution
+    attn_implementation='eager'
 )
 tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-4k-instruct")
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
 def chatbot_response(user_input):
-    """ Generates a response from the chatbot based on the user's input. """
     messages = [{"role": "user", "content": user_input}]  # Create a fresh conversation context
     output = pipe(messages, max_new_tokens=500, return_full_text=False, temperature=0.0, do_sample=False)
     return output[0]['generated_text']
 
-# Define the interface
 iface = gr.Interface(
     fn=chatbot_response,
-    inputs=gr.inputs.Textbox(lines=2, placeholder="Type your question here..."),
-    outputs="text",
+    inputs=gr.components.Textbox(lines=2, placeholder="Type your question here..."),
+    outputs=gr.components.Text(label="Response"),
     title="Smart Chatbot",
     description="This is a smart chatbot that can answer your questions. Just type your question below and get an instant response.",
-    theme="huggingface",  # Apply Hugging Face theme
-    css=".output_text {color: blue; font-size: 16px; font-weight: bold;}"  # Additional custom CSS for styling
+    theme="huggingface"
 )
 
 if __name__ == "__main__":