Update app.py
Browse files
app.py
CHANGED
|
@@ -63,7 +63,7 @@ model = EfficientNetb0_TransformerGLAM(
|
|
| 63 |
)
|
| 64 |
|
| 65 |
# ✅ Load the trained checkpoint
|
| 66 |
-
model.load_state_dict(torch.load('
|
| 67 |
model.eval()
|
| 68 |
|
| 69 |
# ✅ Class Names
|
|
|
|
| 63 |
)
|
| 64 |
|
| 65 |
# ✅ Load the trained checkpoint
|
| 66 |
+
model.load_state_dict(torch.load('efficientnet_glam_best.pt', map_location='cpu'))
|
| 67 |
model.eval()
|
| 68 |
|
| 69 |
# ✅ Class Names
|