Update app.py

app.py

@@ -38,14 +38,22 @@ def generate_response(input_text):
     output = pipe(messages, **generation_args)
     return output[0]['generated_text']
 
-# Membuat antarmuka menggunakan Gradio
-iface = gr.Interface(
-    fn=generate_response,             # Fungsi untuk menangani input
-    inputs=gr.Textbox(label="Ask me anything!", placeholder="Tanyakan sesuatu..."),  # Input teks
-    outputs=gr.Textbox(label="AI Response"),  # Output teks dari AI
-    title="AI Chatbot Assistant",      # Judul aplikasi
-    description="Tanya apapun, saya siap membantu!",  # Deskripsi aplikasi
-)
+# Membuat antarmuka menggunakan Gradio dengan Blocks
+with gr.Blocks() as demo:
+    # Pesan pemberitahuan jika menggunakan CPU
+    gr.Markdown(
+        "### ⚠ Sorry for the inconvenience. The Space is currently running on the CPU, which might affect performance. We appreciate your understanding."
+    )
+    
+    # Gradio Interface dengan layout diatur dan pesan di atas
+    gr.Interface(
+        fn=generate_response,             # Fungsi untuk menangani input
+        inputs=gr.Textbox(label="Ask me anything!", placeholder="Tanyakan sesuatu..."),  # Input teks
+        outputs=gr.Textbox(label="AI Response", lines=10),  # Output teks dari AI, menampilkan lebih banyak baris
+        title="AI Chatbot Assistant",      # Judul aplikasi
+        description="Tanya apapun, saya siap membantu!",  # Deskripsi aplikasi
+        layout="vertical"  # Layout output di atas input
+    ).render()
 
 # Menjalankan antarmuka
-iface.launch()
+demo.launch()