Update app.py
Browse files
app.py
CHANGED
@@ -58,8 +58,8 @@ def generate_image(seed_image):
|
|
58 |
_, file_ext = os.path.splitext(seed_image)
|
59 |
|
60 |
if file_ext.lower() == '.jp2':
|
61 |
-
input_img = Image.
|
62 |
-
input_img_pil =
|
63 |
elif file_ext.lower() == '.fits':
|
64 |
with fits.open(seed_image) as hdul:
|
65 |
data = hdul[0].data
|
|
|
58 |
_, file_ext = os.path.splitext(seed_image)
|
59 |
|
60 |
if file_ext.lower() == '.jp2':
|
61 |
+
input_img = Image.open(seed_image)
|
62 |
+
input_img_pil = transform_hmi_jp2(input_img).reshape(1, 1, 256, 256).to(device)
|
63 |
elif file_ext.lower() == '.fits':
|
64 |
with fits.open(seed_image) as hdul:
|
65 |
data = hdul[0].data
|