add model weights

app.py

@@ -30,6 +30,7 @@ print("Loading resnet...")
 model = resnet50(weights=ResNet50_Weights.IMAGENET1K_V2)
 hidden_state_size = model.fc.in_features
 model.fc = torch.nn.Linear(in_features=hidden_state_size, out_features=2, bias=True)
+model.load_state_dict(torch.load("model_best_epoch_4_59.62.pth"))
 model.to("cuda")
 
 import gradio as gr

model_best_epoch_4_59.62.pth

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