fix: return number instead of text

X_test.npy

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:e33b145f1d8644611b497666ffaac86b3bed0b928dcf9fd8ab7e06216c754591
-size 3840128

app.py

@@ -14,16 +14,13 @@ data = np.load("X_test.npy")
 iso_forest = IsolationForest(contamination=0.1, random_state=42)
 iso_forest.fit(data)
 
-def test_email(text):
+def classify_email(text):
     with torch.no_grad():
         inputs = tokenizer(text, return_tensors='pt', truncation=True, padding=True, max_length=256)
         outputs = model(**inputs)
         cls_embedding = outputs.last_hidden_state[:, 0, :].cpu().numpy()
     pred = iso_forest.predict(cls_embedding)[0]
-    if pred == -1:
-        return "Anomaly detected"
-    else:
-        return "Normal"
+    return pred
 
-demo = gr.Interface(fn=test_email, inputs="text", outputs="text")
+demo = gr.Interface(fn=classify_email, inputs="text", outputs="number")
 demo.launch()

requirements.txt

@@ -3,4 +3,4 @@ scikit-learn
 tqdm
 torch
 gradio
-numpy
+numpy