Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load lightweight model to avoid memory issues (you can replace it with Falcon or others)
+# Load a lightweight open-access model
 generator = pipeline("text-generation", model="tiiuae/falcon-7b-instruct", device_map="auto", max_new_tokens=200)
 
 def respond(message, chat_history):
@@ -11,15 +11,16 @@ def respond(message, chat_history):
     chat_history.append((message, response))
     return "", chat_history
 
-# Create Gradio app with Submit button
+# Gradio Blocks interface with Submit button
 with gr.Blocks() as demo:
-    gr.Markdown("## 🗨️ Falcon Chatbot with Submit Button")
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox(placeholder="Enter your message here", label="Your Message")
-    submit = gr.Button("Submit")
+    gr.Markdown("## 🤖 Falcon-7B Chatbot with Submit Button")
+    chatbot = gr.Chatbot(label="Chat")
+    with gr.Row():
+        msg = gr.Textbox(placeholder="Type your message here...", show_label=False, scale=8)
+        submit = gr.Button("Submit", scale=2)
     state = gr.State([])
 
-    # Bind button click and Enter key
+    # Bind submit button and Enter key
     submit.click(fn=respond, inputs=[msg, state], outputs=[msg, chatbot])
     msg.submit(fn=respond, inputs=[msg, state], outputs=[msg, chatbot])