Update app.py

app.py

@@ -6,6 +6,8 @@ import cv2 as cv
 import numpy as np
 import onnxruntime
 
+from PIL import Image
+
 import gradio
 
 def run_inference(onnx_session, input_size, image):
@@ -39,6 +41,7 @@ def run_inference(onnx_session, input_size, image):
 onnx_session = onnxruntime.InferenceSession("model_float32.onnx")
 
 def create_rgba(image):
+  print(images.shape)
   return run_inference(
       onnx_session,
       [480,640],