Plsek commited on
Commit
5a26856
·
1 Parent(s): 91184b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -9,12 +9,12 @@ from astropy.wcs import WCS
9
  from astropy.nddata import Cutout2D, CCDData
10
 
11
  # Tensorflow
12
- from tensorflow.keras.models import load_model
13
- model = load_model("CADET.hdf5")
14
 
15
  # HuggingFace
16
- # from huggingface_hub import from_pretrained_keras
17
- # model = from_pretrained_keras("Plsek/CADET-v1")
18
 
19
  # Streamlit
20
  import streamlit as st
 
9
  from astropy.nddata import Cutout2D, CCDData
10
 
11
  # Tensorflow
12
+ # from tensorflow.keras.models import load_model
13
+ # model = load_model("CADET.hdf5")
14
 
15
  # HuggingFace
16
+ from huggingface_hub import from_pretrained_keras
17
+ model = from_pretrained_keras("Plsek/CADET-v1")
18
 
19
  # Streamlit
20
  import streamlit as st