add model path

app.py

@@ -9,9 +9,10 @@ import os
 import gradio as gr
 import albumentations as A
 from albumentations.pytorch import ToTensorV2
-import requests
-import random
+import urllib.request
 
+MODEL_URL = "https://huggingface.co/caisarl76/HI_motorcycle_trunk_cls_model/resolve/main/best_model.pth"
+MODEL_PATH = "/workspace/result/best_model.pth" 
 
 test_transforms = A.Compose(
     [
@@ -33,13 +34,14 @@ def predict(img):
 
 labels = ['no_trunk', 'trunk']
 
-model = models.resnet50(pretrained=True)
+model = models.resnet50(pretrained=False)
 model.fc = nn.Sequential(
     nn.Dropout(0.5),
     nn.Linear(model.fc.in_features, 2)
 )
-MODEL_PATH=''
-# model.load_state_dict(torch.load(MODEL_PATH))
+
+urllib.request.urlretrieve(MODEL_URL, MODEL_PATH)
+model.load_state_dict(torch.load(MODEL_PATH))
 model.eval()
 
 inputs = gr.inputs.Image()