Update app.py

app.py

@@ -53,7 +53,7 @@ def inference(image_path_or_url, learn):
     im_new.save("test.jpg", quality=95)
     img = open_image("test.jpg")
     p, img_hr, b = learn.predict(img)
-    return img_hr
+    return img_hr.to_pil()
 
 # Streamlit application
 st.title("Image Inference with Fastai")