SakshiRathi77 commited on
Commit
8429a55
1 Parent(s): 97ff8bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,7 +1,8 @@
1
  import gradio as gr
2
  import spaces
3
  from huggingface_hub import hf_hub_download
4
-
 
5
 
6
  # def download_models(model_id):
7
  # hf_hub_download("SakshiRathi77/void-space-detection/weights", filename=f"{model_id}", local_dir=f"./")
@@ -20,8 +21,7 @@ def yolov9_inference(img_path, image_size, conf_threshold, iou_threshold):
20
  :param size: Optional, input size for inference.
21
  :return: A tuple containing the detections (boxes, scores, categories) and the results object for further actions like displaying.
22
  """
23
- # Import YOLOv9
24
- import yolov9
25
 
26
  # Load the model
27
  # model_path = download_models()
 
1
  import gradio as gr
2
  import spaces
3
  from huggingface_hub import hf_hub_download
4
+ # Import YOLOv9
5
+ import yolov9
6
 
7
  # def download_models(model_id):
8
  # hf_hub_download("SakshiRathi77/void-space-detection/weights", filename=f"{model_id}", local_dir=f"./")
 
21
  :param size: Optional, input size for inference.
22
  :return: A tuple containing the detections (boxes, scores, categories) and the results object for further actions like displaying.
23
  """
24
+
 
25
 
26
  # Load the model
27
  # model_path = download_models()