Update app.py

app.py

@@ -71,7 +71,7 @@ def generate_ocr(method, img):
         probs = F.softmax(outputs.logits, dim=1)  # Convert logits to probabilities
         prediction = torch.argmax(probs, dim=1).item()
 
-    label_map = {0: "Not Spam", 1: "Spam"}
+    label_map = {0: "Spam", 1: "Not Spam"}
     label = label_map[prediction]
 
     # Save results using the external save function