Spaces:

Kathir0011
/

Birds_Classification

Sleeping

Kathir commited on Mar 24, 2023

Commit

68f7e9f

1 Parent(s): 44d3940

Added project files

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ vit_b16_model, vit_transforms = create_vit_b16(num_classes=len(class_names))
 vit_b16_model.load_state_dict(
     torch.load(
-        f="/content/Birds_Classification/ViT_B16_510_classes.pth",
         map_location=torch.device("cpu")
     )
 )

 vit_b16_model.load_state_dict(
     torch.load(
+        f="ViT_B16_510_classes.pth",
         map_location=torch.device("cpu")
     )
 )