Update app.py

app.py

@@ -8,7 +8,7 @@ def inference(input_img):
     temp = uuid.uuid4()
     shell = f"python yolov9/detect.py --source {input_img} --img 640 --device cpu --weights yolov9/runs/train/exp/weights/best.pt --name {temp}"
     os.system(shell)
-    return f"yolov9/runs/detect/{temp}/{input_img.split('/')[-1]}"
+    return Image.open(f"yolov9/runs/detect/{temp}/{input_img.split('/')[-1]}")
     #return "yolov9/runs/detect/f807164a-496b-413c-bb47-f5daf8803dcd/cut_a_1.mp4"
 
 def inference_video(input_img):
@@ -43,7 +43,7 @@ with gr.Blocks() as demo:
         )
         with gr.Row():
             img_input = [gr.Image(type="filepath",label="Input Image",width=300, height=300)]
-            pred_outputs = [gr.Image(label="Output Image",width=640, height=640)]
+            pred_outputs = [gr.Image(type="PIL",label="Output Image",width=640, height=640)]
         
         image_button = gr.Button("Predict")
         image_button.click(inference, inputs=img_input, outputs=pred_outputs)