Spaces:

Edesak
/

FoodVisionMini

Sleeping

Edesak commited on Jun 1, 2023

Commit

6a31e2f

1 Parent(s): 7f11020

Fixing load function

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ article = "My Github page [Edesak](https://github.com/Edesak)"
 class_names = ["pizza", "steak", "sushi"]
 model,transform = create_model()
-model.load_state_dict(torch.load(f="_Deploy_effB2.pth"),map_location=torch.device('cpu'))
 model.to("cpu")
 example_list = [["Examples/" + example] for example in os.listdir("Examples")]

 class_names = ["pizza", "steak", "sushi"]
 model,transform = create_model()
+model.load_state_dict(torch.load(f="_Deploy_effB2.pth",map_location=torch.device('cpu')))
 model.to("cpu")
 example_list = [["Examples/" + example] for example in os.listdir("Examples")]