Update app.py

app.py

@@ -4,7 +4,7 @@ import joblib
 # Function to load your model (adjust the path and method if needed)
 def load_model():
     # This path is relative to the root of your Hugging Face Space
-    model_path = "./en-hate-speech-detection-3label"
+    model_path = "./random_forest_model_3labels2.joblib"
     model = joblib.load(model_path)
     return model
 
@@ -20,7 +20,7 @@ def predict_hate_speech(text):
 # Adjusted Gradio interface to take text input and output model predictions
 iface = gr.Interface(fn=predict_hate_speech, 
                      inputs=gr.Textbox(lines=2, placeholder="Enter Text Here..."), 
-                     outputs="text", 
+                     outputs=labels[prediction[0]], 
                      description="Detects hate speech in text. Outputs 'Neutral or Ambiguous', 'Not Hate', 'Offensive or Hate Speech'.")
 iface.launch()