Update app.py

app.py

@@ -12,13 +12,22 @@ def classify_text(text):
  result = classifier(text)
  return result[0]['label'], result[0]['score']
 
-with gr.Blocks() as demo:
- gr.Markdown("# Hate Speech Classification with Fine-Tuned Model")
- with gr.Row():
- text_input = gr.Textbox(label="Input Text")
- label_output = gr.Textbox(label="Predicted Label")
- score_output = gr.Number(label="Prediction Score")
+# with gr.Blocks() as demo:
+# gr.Markdown("# Hate Speech Classification with Fine-Tuned Model")
+# with gr.Row():
+# text_input = gr.Textbox(label="Input Text")
+# label_output = gr.Textbox(label="Predicted Label")
+# score_output = gr.Number(label="Prediction Score")
  
- text_input.change(fn=classify_text, inputs=text_input, outputs=[label_output, score_output])
+# text_input.change(fn=classify_text, inputs=text_input, outputs=[label_output, score_output])
 
-demo.launch()
+# demo.launch()
+
+iface = gr.Interface(
+ fn=classify_text,
+ inputs=gr.Textbox(label="Input Text"),
+ outputs=[gr.Textbox(label="Predicted Label"), gr.Number(label="Prediction Score")],
+ title="Hate Speech Classification with Fine-Tuned Model"
+)
+
+iface.launch()