Update app.py

app.py

@@ -26,7 +26,7 @@ def yolov7_inference(
         Rendered image
     """
 
-    model = torch.hub.load('kadirnar/yolov7-v0.1', 'custom', path='skin_burn.pt', source='local', device="cpu")
+    model = torch.hub.load('kadirnar/yolov7-v0.1', 'custom', path='skin_burn.pt', source='local', hf_model=True, device="cpu")
     model.conf = conf_threshold
     model.iou = iou_threshold
     results = model([image], size=image_size)
@@ -38,7 +38,7 @@ inputs = [
     gr.inputs.Dropdown(
         choices=[
             "skin_burn",
-            "kadirnar/yolov7-v0.1",
+            "Other"
         ],
         default="skin_burn",
         label="Model",