Update app.py

app.py

@@ -29,7 +29,7 @@ NUM_CLASSES = 2
 CLASS_NAMES = ["Elliptical", "Spiral"]
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-# 🔴 Calibration + OOD thresholds
+# Calibration + OOD thresholds
 TEMPERATURE = 2.5          # softens overconfidence
 CONF_THRESHOLD = 0.60      # below this → uncertain
 ENTROPY_THRESHOLD = 0.85   # high entropy → uncertain
@@ -52,7 +52,7 @@ def load_model():
     if os.path.exists(MODEL_PATH):
         try:
             state_dict = torch.load(MODEL_PATH, map_location=DEVICE)
-            model.load_state_dict((torch.load(MODEL_PATH, map_location=DEVICE))
+            model.load_state_dict((torch.load(MODEL_PATH, map_location=DEVICE)))
             print(f"Model loaded successfully from {MODEL_PATH}")
         except Exception as e:
             print(f"Error loading model: {e}")