Update app.py

app.py

@@ -3,13 +3,13 @@ from setfit import SetFitModel
 
 model = SetFitModel.from_pretrained("nickprock/setfit-italian-hate-speech")
 
-dictionary={'0':'not-hateful',
-'1':'hateful',}
+dictionary={0:'not-hateful',
+1:'hateful',}
 
 
 def greet(text, dictionary=dictionary):
   preds = model(text)
-  return response
+  return dictionary[response[0]][label]
 
 demo = gr.Interface(fn=greet, 
   inputs="text",