Spaces:
Sleeping
Sleeping
SakshiRathi77
commited on
Commit
•
8429a55
1
Parent(s):
97ff8bc
Update app.py
Browse files
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 |
-
|
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()
|