Update app.py

app.py

@@ -3,11 +3,11 @@ from huggingface_hub.utils import HfHubHTTPError
 
 def predict(message, history):
     try:
-        # Define the model interface directly using the load method
-        chat_interface = gr.ChatInterface.load("models/meta-llama/Meta-Llama-3.1-8B")
+        # Load the model interface directly using the load method
+        chat_interface = gr.Interface.load("models/meta-llama/Meta-Llama-3.1-8B")
 
-        # Use the interface to get predictions
-        response = chat_interface(message, history)
+        # Get the prediction using the correct method call
+        response = chat_interface.predict(message)
         history.append((message, response))
         return "", history
 
@@ -23,5 +23,3 @@ with gr.Blocks() as demo:
     clear = gr.ClearButton([msg, chatbot])
 
     msg.submit(predict, [msg, chatbot], [msg, chatbot])
-
-demo.launch()