Update app.py

app.py

@@ -5,7 +5,12 @@ from torch.nn import functional as F
 import torchvision
 from torchvision import transforms
 
-model = torch.hub.load('ultralytics/yolov5', 'custom', 'runs/train-cls/exp/weights/best.pt')
+REPO_ID = "Kr1n3/Fashion-Items-Classification"
+FILENAME = "best.pt"
+
+yolov5_weights = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
+
+model = torch.hub.load('ultralytics/yolov5', 'custom', path=yolov5_weights, force_reload=True)
 data_transform1=transforms.Compose([
     transforms.Resize((224,224)),
     transforms.ToTensor(),