Update app.py

app.py

@@ -44,10 +44,11 @@ def generate_image(seed_image):
     elif file_ext.lower() == '.fits':
         with fits.open(seed_image) as hdul:
             data = hdul[0].data
-            
             input_img_pil = transform_hmi(data).reshape(1, 1, 256, 256).to(device)
+    else:
+        print(f'Format {file_ext.lower()} not supported')
             
-    generated_image = diffusion.sample(model, y=seed_image_tensor, labels=None, n=1)
+    generated_image = diffusion.sample(model, y=input_img_pil, labels=None, n=1)
 
     inp_img = seed_image_tensor.reshape(1, 256, 256).permute(1, 2, 0)  
     inp_img = np.squeeze(inp_img.cpu().numpy())