malavikapradeep2001 commited on
Commit
a16a1ba
·
unverified ·
1 Parent(s): 5b3b9fe

Update backend/app.py

Browse files
Files changed (1) hide show
  1. 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 = ['neg', 'pos']
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