Spaces:
Sleeping
Sleeping
Update backend/app.py
Browse files- backend/app.py +1 -1
backend/app.py
CHANGED
|
@@ -68,7 +68,7 @@ print("🔹 Loading MWT model...")
|
|
| 68 |
mwt_model = create_model(num_classes=2).to(device)
|
| 69 |
mwt_model.load_state_dict(torch.load("MWTclass2.pth", map_location=device))
|
| 70 |
mwt_model.eval()
|
| 71 |
-
mwt_class_names = ['
|
| 72 |
|
| 73 |
# =====================================================
|
| 74 |
# Model 3: CIN Classifier
|
|
|
|
| 68 |
mwt_model = create_model(num_classes=2).to(device)
|
| 69 |
mwt_model.load_state_dict(torch.load("MWTclass2.pth", map_location=device))
|
| 70 |
mwt_model.eval()
|
| 71 |
+
mwt_class_names = ['Negative', 'Positive']
|
| 72 |
|
| 73 |
# =====================================================
|
| 74 |
# Model 3: CIN Classifier
|