Update app.py

app.py

@@ -27,6 +27,7 @@ def fn(image):
     im = load_img(image, output_type="pil")
     im = im.convert("RGB")
     image_size = im.size
+    origin = im.copy()
     input_images = transform_image(im).unsqueeze(0).to("cpu")
     
     with torch.no_grad():
@@ -48,6 +49,7 @@ def fn_url(url):
     im = load_img(url, output_type="pil")
     im = im.convert("RGB")
     image_size = im.size
+    origin = im.copy()
     input_images = transform_image(im).unsqueeze(0).to("cpu")
     
     with torch.no_grad():