Spaces:

nickprock
/

italian-hate-speech

Sleeping

nickprock commited on Mar 23, 2023

Commit

7ad2f16

•

1 Parent(s): 5d46de5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,8 +8,8 @@ dictionary={0:'not-hateful',
 def greet(text, dictionary=dictionary):
-  response = model(text)
-  return dictionary[response[0]][label]
 demo = gr.Interface(fn=greet,
   inputs="text",

 def greet(text, dictionary=dictionary):
+  response = model([text])
+  return dictionary[response[0].item()]
 demo = gr.Interface(fn=greet,
   inputs="text",