Update app.py(1)

app.py

@@ -5,11 +5,12 @@ model_id = "Subhajit01/spam_detector_powered_by_custom_GPT"
 classifier = pipeline("text-classification", model=model_id, tokenizer="gpt2")
 labels = ["spam", "not spam"]
 
-def classify(result):
-  if result[0]["label"] == 'LABEL_0':
-    print(labels[0])
-  else:
-    print(labels[1])
+def classify(text):
+    result = classifier(text)
+    if result[0]["label"] == 'LABEL_0':
+        return labels[0]
+    else:
+        return labels[1]
       
 with gr.Blocks() as demo:
     gr.Markdown("# Spam Detection System")