model commit

app.py

@@ -25,7 +25,7 @@ model  = create_model(
 # Load saved weights
 model.load_state_dict(
     torch.load(
-        f="09_pretrained_effnetb2_feature_extractor_pizza_steak_sushi_20_percent.pth",
+        f="cnn-resnet-version-1.pt",
         map_location=torch.device("cpu"),  # load to CPU
     )
 )

cnn-resnet-version-1.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e78681c1005db8327d6ef33ce67a40abb2c57723ef77f1ccc73fcf7e59ed20e6
+size 98552625

requirements.txt

@@ -1,3 +1,3 @@
-torch==1.12.0
-torchvision==0.13.0
-gradio==3.1.4
+torch
+torchvision
+gradio