Spaces:

citradiani
/

food-tiny-classification

Running

citradiani commited on Mar 20, 2024

Commit

4d69d16

verified ·

1 Parent(s): d81bd61

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ def predict(img_path):
     # Apply transformations
     image = test_transforms(image)
-    inf_model = models.resnet18(pretrained=False, weights='ResNet50_Weights.DEFAULT')
     num_ftrs = inf_model.fc.in_features
     # Here the size of each output sample is set to 2.

     # Apply transformations
     image = test_transforms(image)
+    inf_model = models.resnet18(pretrained=False)
     num_ftrs = inf_model.fc.in_features
     # Here the size of each output sample is set to 2.