Tirath5504 commited on
Commit
9dc50fd
1 Parent(s): 7bd4481

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def predict(text):
49
  prediction = "The string is classified as hate speech."
50
  else:
51
  prediction = "The string is classified as normal speech."
52
- return {"prediction": prediction, "probability": probabilities.tolist()}
53
 
54
  iface = gr.Interface(
55
  fn=predict,
 
49
  prediction = "The string is classified as hate speech."
50
  else:
51
  prediction = "The string is classified as normal speech."
52
+ return prediction, probabilities.tolist()
53
 
54
  iface = gr.Interface(
55
  fn=predict,