Update app.py

app.py

@@ -1,9 +1,14 @@
+import torch
+import argparse
 import gradio as gr
+from PIL import Image
+from numpy import random
+from pathlib import Path
 import os
-import torch
+import time
 import torch.backends.cudnn as cudnn
 from models.experimental import attempt_load
-
+import cv2
 from utils.datasets import LoadStreams, LoadImages
 from utils.general import check_img_size, check_requirements, check_imshow, non_max_suppression, apply_classifier,scale_coords, xyxy2xywh, strip_optimizer, set_logging, increment_path
 from utils.plots import plot_one_box
@@ -12,10 +17,10 @@ from utils.torch_utils import select_device, load_classifier, time_synchronized,
 os.system('git clone https://github.com/WongKinYiu/yolov7')
 
 
-def detect(inp):
+'''def detect(inp):
   os.system('python ./yolov7/detect.py --weights best.pt --conf 0.25 --img-size 640 --source f{inp} --project ./yolov7/runs/detect ')
   otp=inp.split('/')[2]
-  return f"./yolov7/runs/detect/exp/*"
+  return f"./yolov7/runs/detect/exp/*"'''
    
 
   
@@ -53,10 +58,8 @@ def Custom_detect(img,model):
     img.save("Temp_file/test.jpg")
     source, weights, view_img, save_txt, imgsz, trace = opt.source, opt.weights, opt.view_img, opt.save_txt, opt.img_size, opt.trace
     save_img = True 
-    webcam = source.isnumeric() or source.endswith('.txt') or source.lower().startswith(
-        ('rtsp://', 'rtmp://', 'http://', 'https://'))
-    save_dir = Path(increment_path(Path(opt.project) / opt.name, exist_ok=opt.exist_ok)) 
-    (save_dir / 'labels' if save_txt else save_dir).mkdir(parents=True, exist_ok=True)  
+    webcam = source.isnumeric() or source.endswith('.txt') or source.lower().startswith(('rtsp://', 'rtmp://', 'http://', 'https://'))
+    save_dir = Path(increment_path(Path(opt.project) / opt.name, exist_ok=opt.exist_ok))(save_dir / 'labels' if save_txt else save_dir).mkdir(parents=True, exist_ok=True)  
     set_logging()
     device = select_device(opt.device)
     half = device.type != 'cpu'