changed app

.gitignore

@@ -1,5 +1,4 @@
 flagged/
 *.pt
 *.png
-*.jpg
 gradio_cached_examples/

app.py

@@ -6,27 +6,12 @@ import os
 import torch
 import ultralytics
  
-file_urls = [
-    'https://www.dropbox.com/s/bc9r8n7919cbc77/test-image.jpg?dl=0',
-    'https://www.dropbox.com/s/fkmzgdm6okdzxdk/test-image-2.jpg?dl=0',
-]
- 
-def download_file(url, save_name):
-    url = url
-    if not os.path.exists(save_name):
-        file = requests.get(url)
-        open(save_name, 'wb').write(file.content)
- 
-for i, url in enumerate(file_urls):
-    download_file(
-        file_urls[i],
-        f"image_{i}.jpg"
-    )
 
 model = torch.hub.load("ultralytics/yolov5", "custom", path="yolov5_0.65map_exp7_best.pt",
                        force_reload=False) 
+model.conf = 0.20  # NMS confidence threshold
 
-path  = [['image_0.jpg'], ['image_1.jpg']]
+path  = [['img/test-image.jpg'], ['img/test-image-2']]
 
 def show_preds_image(image_path):
     image = cv2.imread(image_path)
@@ -42,7 +27,9 @@ def show_preds_image(image_path):
             lineType=cv2.LINE_AA
         )
     return cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
- 
+
+
+
 inputs_image = [
     gr.components.Image(type="filepath", label="Input Image"),
 ]