Create chatbot.py

chatbot.py

@@ -0,0 +1,13 @@
+import gradio as gr
+from gemma_model import generate_text
+
+chatbot = gr.Chatbot(
+    generate_text,
+    title="Gemma-2B-IT Chatbot",
+    description="Talk to the Gemma-2B-IT model!",
+    width=800,
+    height=600,
+)
+
+if __name__ == "__main__":
+    chatbot.launch()