Rename adanalyst_mar30_logo (1).py to app.py

adanalyst_mar30_logo (1).py → app.py

@@ -477,11 +477,12 @@ def perform_object_detection(image_pil):
 """## Logo"""
 
 # Commented out IPython magic to ensure Python compatibility.
-!git clone https://github.com/WongKinYiu/yolov7.git
-# %cd yolov7
-!wget -O logo_detection.pt https://huggingface.co/spaces/xktan/Ad_Analyst/resolve/main/logo_detection.pt
-
-logo_model = torch.hub.load('/content/yolov7', 'custom', '/content/yolov7/logo_detection.pt', source='local')
+if not os.path.exists("yolov7"):
+    with zipfile.ZipFile("yolov7.zip", 'r') as zip_ref:
+        zip_ref.extractall(".")
+print("files", os.listdir("."))
+sys.path.append("./yolov7")
+logo_model = torch.hub.load('./yolov7', 'custom', './logo_detection.pt', source='local')
 logo_model.conf = 0.25
 
 def detect_logos(pil_image):