Update app.py

app.py

@@ -7,14 +7,14 @@ def predict(message, history):
         # Create the Interface object directly
         iface = gr.Interface(
             fn=gr.ChatInterface.load("models/meta-llama/Meta-Llama-3.1-8B"),
-            inputs="textbox",
-            outputs="textbox",
+            chatbot=gr.Chatbot(),  # Use gr.Chatbot() directly
+            textbox=gr.Textbox(),
         )
 
         # Launch the interface and get the prediction function
         with iface:
             fn = iface.predict
-        
+
         response = fn(message, history)
         history.append((message, response))
         return "", history
@@ -26,7 +26,9 @@ def predict(message, history):
             raise e
 
 
-with gr.Blocks() as demo:
+demo = gr.Blocks()
+
+with demo:
     chatbot = gr.Chatbot()
     msg = gr.Textbox()
     clear = gr.ClearButton([msg, chatbot])