Update app.py

app.py

@@ -8,13 +8,9 @@ from astropy.io import fits
 from astropy.wcs import WCS
 from astropy.nddata import Cutout2D, CCDData
 
-# Tensorflow
-from tensorflow.keras.models import load_model
-model = load_model("CADET.hdf5")
-
 # HuggingFace
-# from huggingface_hub import from_pretrained_keras
-# model = from_pretrained_keras("Plsek/CADET-v1")
+from huggingface_hub import from_pretrained_keras
+model = from_pretrained_keras("Plsek/CADET-v1")
 
 # Streamlit
 import streamlit as st