Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -107,7 +107,7 @@ def yolov9_inference(model_id, img_path=None, vid_path=None, tracking_algorithm
|
|
107 |
|
108 |
plt.tight_layout() # Ensure the entire plot fits into the figure area
|
109 |
#ax.set_facecolor('#D3D3D3')
|
110 |
-
|
111 |
output_video = output_path # Load the video file here
|
112 |
output_image = None
|
113 |
plt.style.use("ggplot")
|
|
|
107 |
|
108 |
plt.tight_layout() # Ensure the entire plot fits into the figure area
|
109 |
#ax.set_facecolor('#D3D3D3')
|
110 |
+
elif output_extension.lower() in vid_extensions:
|
111 |
output_video = output_path # Load the video file here
|
112 |
output_image = None
|
113 |
plt.style.use("ggplot")
|