Update app.py

app.py

@@ -15,6 +15,7 @@ def init():
         #raise ValueError("HUGGINGFACE_TOKEN environment variable is not set")
     
     model_id = "meta-llama/Meta-Llama-3.1-8B-Instruct"
+    model_id = "google/gemma-2b"
     
     device = "auto" # torch.device("cuda" if torch.cuda.is_available() else "cpu")
     #device = "cuda"
@@ -87,7 +88,9 @@ def call_generate_text(message, history):
 js = """
 console.log('hello js')
 """
-demo = gr.ChatInterface(call_generate_text,type="messages")
+chatbot = gr.Chatbot()
+
+demo = gr.ChatInterface(call_generate_text,chatbot=chatbot,type="messages")
 
 if __name__ == "__main__":
     init()