Spaces:

Sandhya2002
/

Food_Vision_Mini

Sleeping

Sandhya commited on Apr 25, 2025

Commit

e1d34a8

1 Parent(s): 116a05c

change app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ from model import create_model
 from timeit import default_timer as Timer
 class_names=["Pizza","Steak","Sushi"]
 effnetb2,effnetb2_transform=create_model(num_class=3)
-effnetb2.load_state_dict(torch.load(f="demos\foodvision_mini\effnetb2_pizza_steak_sushi (1).pt",map_location=torch.device('cpu')))
 def predict(img):

 from timeit import default_timer as Timer
 class_names=["Pizza","Steak","Sushi"]
 effnetb2,effnetb2_transform=create_model(num_class=3)
+effnetb2.load_state_dict(torch.load(f="effnetb2_pizza_steak_sushi (1).pt",map_location=torch.device('cpu')))
 def predict(img):