Spaces:

kietnt0603
/

ViT_Food101_Demo

Sleeping

kietnt0603 commited on Nov 16, 2023

Commit

609d43b

1 Parent(s): c4a1f55

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ model, transform = create_model(
 # Load saved weights
 model.load_state_dict(
     torch.load(
-        f="/content/drive/MyDrive/CS431-EndCourseProject/vit_b_16_food101_cifar100.pth",
         map_location=torch.device("cpu"),  # load to CPU
     )
 )

 # Load saved weights
 model.load_state_dict(
     torch.load(
+        f="vit_b_16_food101_cifar100.pth",
         map_location=torch.device("cpu"),  # load to CPU
     )
 )