Spaces:
Sleeping
Sleeping
Tirath5504
commited on
Commit
•
9dc50fd
1
Parent(s):
7bd4481
Update app.py
Browse files
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
|
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,
|