Update app.py

app.py

@@ -33,7 +33,26 @@ def yolov8_inference(
     render = render_result(model=model, image=image, result=results[0])
     
     return render
-        
+
+def gr_postprocess(image):
+    """
+    Gradio postprocess function
+    Args:
+        image: Input image
+    Returns:
+        Processed image
+    """
+
+    # Convert the image to RGB
+    image = image.convert('RGB')
+
+    # Resize the image to the desired size
+    image = image.resize((640, 480))
+
+    # Convert the image to a numpy array
+    image = np.array(image)
+
+    return image
 
 inputs = [
 # Images
@@ -42,7 +61,7 @@ inputs = [
             'samples/1.jpeg',
             'samples/2.JPG',
         ],
-        inputs={'postprocess': gr.postprocess()},
+        inputs={'postprocess': gr_postprocess()},
     ),
     gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
     gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.25, step=0.05, label="Confidence Threshold"),