Update app.py

app.py

@@ -69,9 +69,7 @@ os.system("wget https://github.com/AK391/models/raw/main/vision/super_resolution
 # For example, if NVIDIA GPU is available and ORT Python package is built with CUDA, then call API as following:
 # onnxruntime.InferenceSession(path/to/model, providers=['CUDAExecutionProvider'])
 ort_session = onnxruntime.InferenceSession("super-resolution-10.onnx")
-ort_inputs = {ort_session.get_inputs()[0].name: img_5} 
-ort_outs = ort_session.run(None, ort_inputs)
-img_out_y = ort_outs[0]
+
 
 def inference(img):
   orig_img = Image.open(img)
@@ -82,6 +80,11 @@ def inference(img):
   
   img_4 = np.expand_dims(np.expand_dims(img_ndarray, axis=0), axis=0)
   img_5 = img_4.astype(np.float32) / 255.0
+  
+  ort_inputs = {ort_session.get_inputs()[0].name: img_5} 
+  ort_outs = ort_session.run(None, ort_inputs)
+  img_out_y = ort_outs[0]
+  
   img_out_y = Image.fromarray(np.uint8((img_out_y[0] * 255.0).clip(0, 255)[0]), mode='L')
   final_img = Image.merge(
       "YCbCr", [