fix

app.py

@@ -6,10 +6,6 @@ import gradio as gr
 from torchvision.models import resnet50
 from huggingface_hub import hf_hub_download
 
-
-from model import resnet50
-model_path = "checkpoint-epcoh24_lr_point1.pth"
-
 def load_model(model_path):
     model = resnet50(pretrained=False)  # Create an instance of the ResNet-50 model
     state_dict = torch.load(model_path, map_location='cpu')  # Load the state dictionary