Update app.py

app.py

@@ -6,19 +6,8 @@ import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
 def get_hatespeech_score(text):
-    tokenizer = AutoTokenizer.from_pretrained("unhcr/hatespeech-detection")
-    model = AutoModelForSequenceClassification.from_pretrained("unhcr/hatespeech-detection")
-
-    # Tokenize input text
-    inputs = tokenizer(text, return_tensors='pt')
-
-    # Perform inference
-    outputs = model(**inputs)
-
-    # Get predicted label
-    predicted_label_idx = torch.argmax(outputs.logits).item()
-    predicted_label = model.config.id2label[predicted_label_idx]
-    return predicted_label
+    io = gr.Interface.load("unhcr/hatespeech-detection")
+    return io(text)
 
 def text_analysis(text):  
     label_1 = get_hatespeech_score(text)