Spaces:

georgescutelnicu
/

weather-image-classifier

Running

georgescutelnicu commited on Nov 7, 2023

Commit

cceb3ce

1 Parent(s): ff46c8e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ model, transform = create_effnetb2_model(
 model.load_state_dict(
     torch.load(
-        f="model_v4.pth",
         map_location=torch.device("cpu")
     )
 )

 model.load_state_dict(
     torch.load(
+        f="model_v3.pth",
         map_location=torch.device("cpu")
     )
 )