Update app.py

app.py

@@ -29,17 +29,9 @@ def yolov9_inference(img_path, model_path,image_size, conf_threshold, iou_thresh
     results = model(img_path, size=image_size)
 
     # Optionally, show detection bounding boxes on image
-    save_path = 'output/'
-    results.save(labels=True, save_dir=save_path, exist_ok=True)
-    print("save_path:",save_path)
-    print("img_path:",img_path)
-
-    new_image_path = os.path.basename(img_path)
-    output_path = save_path + new_image_path
-    
-    print(f"Output image saved to {output_path}")
+    output = results.render()
     
-    return output_path
+    return output[0]
 
 
 inputs = [
@@ -59,7 +51,7 @@ inputs = [
     gr.Slider(minimum=0.0, maximum=1.0, value=0.45, step=0.05, label="IOU Threshold"),
 ]
 
-outputs = gr.Image(type="filepath",label="Output Image")
+outputs = gr.Image(type="numpy",label="Output Image")
 title = "YOLOv9"
 
 demo_app = gr.Interface(