Spaces:

owaiskha9654
/

Custom_Yolov7

Build error

owaiskha9654 commited on Aug 14, 2022

Commit

d2ef392

•

1 Parent(s): d83d73f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,8 +13,10 @@ try:
     model = torch.hub.load('jinfagang/yolov7', 'custom', path=yolov7_weights, force_reload=True)  # local repo
 except:
     import zipfile
-    with zipfile.ZipFile("../.cache/torch/hub/main.zip","r") as zip_ref:
-        zip_ref.extractall("../.cache/torch/hub/jinfagang_yolov7_main/")
 def object_detection(im, size=416):
     results = model(im)  # inference
     #results.print()  # print results to screen

     model = torch.hub.load('jinfagang/yolov7', 'custom', path=yolov7_weights, force_reload=True)  # local repo
 except:
     import zipfile
+    l_files = os.listdir("./.cache/torch/hub")
+    print(l_files)
+    with zipfile.ZipFile("./.cache/torch/hub/main.zip","r") as zip_ref:
+        zip_ref.extractall("./.cache/torch/hub/jinfagang_yolov7_main/")
 def object_detection(im, size=416):
     results = model(im)  # inference
     #results.print()  # print results to screen