Update app.py

app.py

@@ -6,6 +6,7 @@ import torch
 import torchvision.transforms as transforms
 import requests
 import numpy as np
+from astropy.io import fits
 
 # Preprocessing
 from modules import PaletteModelV2