Hugging Face's logo

Spaces:
morinop
/
JailEval
Sleeping

App Files Community
morinop commited on
Commit
db57a7d
·
1 Parent(s): c71389b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def predict(sequence):
30
  # print("Predicted probabilities:", predictions)
31
  # print("Predicted classes:", predicted_classes)
32
 
33
- return predicted_classes[0]
34
 
35
 
36
  iface = gr.Interface(fn=predict, inputs="text", outputs="text")
 
30
  # print("Predicted probabilities:", predictions)
31
  # print("Predicted classes:", predicted_classes)
32
 
33
+ return predicted_classes[0].item()
34
 
35
 
36
  iface = gr.Interface(fn=predict, inputs="text", outputs="text")