Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import joblib
 
 # Load your serialized objects
 model = joblib.load('random_forest_model_3labels2.joblib')
@@ -14,7 +15,7 @@ def predict(input_text):
     prediction = model.predict(vectorized_text)
     
     # If your model's output needs to be decoded (optional)
-    # decoded_prediction = encoder.inverse_transform(prediction)
+    decoded_prediction = encoder.inverse_transform(prediction)
     
     # Return the prediction (you might want to convert it into a more readable form)
     return prediction[0]  # Modify this according to your needs
@@ -23,7 +24,7 @@ def predict(input_text):
 iface = gr.Interface(fn=predict,
                      inputs=gr.Textbox(lines=2, placeholder="Enter Text Here..."),
                      outputs="text",
-                     description="Your model description here.")
+                     description="Detects hate speech in text. Outputs 'Neutral or Ambiguous', 'Not Hate', or 'Offensive or Hate Speech'.")
 
 # Launch the app
 iface.launch()