Update app.py

app.py

@@ -9,17 +9,7 @@ model = AutoModelForSequenceClassification.from_pretrained(model_name)
 
 @spaces.GPU
 def predict_phishing(text):
-    # Special case handling
-    if "magnificent" in text.lower():
-        return {
-            "Prediction": "✅ Legitimate",
-            "Confidence": "100.00%",
-            "Probability Breakdown": {
-                "Legitimate": "1.0000",
-                "Phishing": "0.0000"
-            }
-        }
-    
+
     model.to('cuda')
     inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=512, padding=True)
     inputs = {k: v.to('cuda') for k, v in inputs.items()}