Update app.py

app.py

@@ -58,8 +58,8 @@ def generate_image(seed_image):
     _, file_ext = os.path.splitext(seed_image)
     
     if file_ext.lower() == '.jp2':
-        input_img = Image.transform_hmi_jp2(seed_image)
-        input_img_pil = transform_hmi(input_img).reshape(1, 1, 256, 256).to(device)
+        input_img = Image.open(seed_image)
+        input_img_pil = transform_hmi_jp2(input_img).reshape(1, 1, 256, 256).to(device)
     elif file_ext.lower() == '.fits':
         with fits.open(seed_image) as hdul:
             data = hdul[0].data