kedimestan
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -31,10 +31,10 @@ model = YOLO(current_model_name)
|
|
31 |
|
32 |
def yolov8_inference(
|
33 |
image: gr.Image = None,
|
34 |
-
model_name:
|
35 |
-
image_size:
|
36 |
-
conf_threshold:
|
37 |
-
iou_threshold:
|
38 |
):
|
39 |
"""
|
40 |
YOLOv8 inference function to return masks and label names for each detected object
|
@@ -110,7 +110,7 @@ examples = [
|
|
110 |
["small-vehicles1.jpeg", "yolov8m-seg.pt", 640, 0.25, 0.45],
|
111 |
]
|
112 |
|
113 |
-
# Build the Gradio demo app
|
114 |
demo_app = gr.Interface(
|
115 |
fn=yolov8_inference,
|
116 |
inputs=inputs,
|
@@ -121,5 +121,11 @@ demo_app = gr.Interface(
|
|
121 |
theme="default",
|
122 |
)
|
123 |
|
124 |
-
# Launch the app
|
125 |
-
demo_app.queue().launch(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
31 |
|
32 |
def yolov8_inference(
|
33 |
image: gr.Image = None,
|
34 |
+
model_name: str = None,
|
35 |
+
image_size: int = 640,
|
36 |
+
conf_threshold: float = 0.25,
|
37 |
+
iou_threshold: float = 0.45,
|
38 |
):
|
39 |
"""
|
40 |
YOLOv8 inference function to return masks and label names for each detected object
|
|
|
110 |
["small-vehicles1.jpeg", "yolov8m-seg.pt", 640, 0.25, 0.45],
|
111 |
]
|
112 |
|
113 |
+
# Build the Gradio demo app with POST functionality
|
114 |
demo_app = gr.Interface(
|
115 |
fn=yolov8_inference,
|
116 |
inputs=inputs,
|
|
|
121 |
theme="default",
|
122 |
)
|
123 |
|
124 |
+
# Launch the app with API-enabled functionality
|
125 |
+
demo_app.queue().launch(
|
126 |
+
enable_queue=True, # Allow for API-style interactions
|
127 |
+
debug=True, # Show detailed errors in case of issues
|
128 |
+
server_name="0.0.0.0", # Host on all IPs
|
129 |
+
server_port=7860, # Custom port for accessing the app
|
130 |
+
share=True # To make the app accessible from a URL
|
131 |
+
)
|