saved with safetensors

09_pretrained_effnetb2_20_percent.safetensors

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

app.py

@@ -4,15 +4,14 @@ import gradio as gr
 
 from typing import Tuple, Dict
 from model import create_effnet2_model
+from safetensors.torch import load_file
 from timeit import default_timer as timer
 
 class_names = ["pizza", "steak", "sushi"]
 
 effnetb2, effnetb2_transforms = create_effnet2_model(num_classes=3)
-effnetb2.load_state_dict(torch.load(
-    f="09_pretrained_effnetb2_20_percent.pth",
-    map_location=torch.device("cpu")
-))
+sd = load_file("09_pretrained_effnetb2_20_percent.safetensors")
+missing, unexpected = effnetb2.load_state_dict(sd, strict=True)
 
 
 

requirements.txt

@@ -1,3 +1,4 @@
 torch==2.8.0
 torchvision==0.23.0
 gradio==5.44.1
+safetensors=0.6.2