Update app.py

app.py

@@ -3,14 +3,11 @@ from huggingface_hub.utils import HfHubHTTPError
 
 def predict(message, history):
     try:
-        # Create the Interface object directly
+        # Define the model interface directly using the load method
         chat_interface = gr.ChatInterface.load("models/meta-llama/Meta-Llama-3.1-8B")
-        
-        # Get the prediction function
-        fn = chat_interface.predict
 
-        # Call the prediction function
-        response = fn(message, history)
+        # Use the interface to get predictions
+        response = chat_interface(message, history)
         history.append((message, response))
         return "", history