Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
# HuggingFace
|
2 |
-
|
3 |
-
|
4 |
|
5 |
# Basic libraries
|
6 |
import numpy as np
|
@@ -33,7 +33,7 @@ with col:
|
|
33 |
st.markdown("Input images should be centered at the centre of the galaxy and point sources should be filled with surrounding background ([dmfilth](https://cxc.cfa.harvard.edu/ciao/ahelp/dmfilth.html)).")
|
34 |
|
35 |
# Create file uploader widget
|
36 |
-
|
37 |
|
38 |
# Define function to plot the uploaded image
|
39 |
def plot_image(image, scale):
|
|
|
1 |
# HuggingFace
|
2 |
+
from huggingface_hub import from_pretrained_keras
|
3 |
+
model = from_pretrained_keras("Plsek/CADET-v1")
|
4 |
|
5 |
# Basic libraries
|
6 |
import numpy as np
|
|
|
33 |
st.markdown("Input images should be centered at the centre of the galaxy and point sources should be filled with surrounding background ([dmfilth](https://cxc.cfa.harvard.edu/ciao/ahelp/dmfilth.html)).")
|
34 |
|
35 |
# Create file uploader widget
|
36 |
+
uploaded_file = st.file_uploader("Choose a FITS file", type=['fits'])
|
37 |
|
38 |
# Define function to plot the uploaded image
|
39 |
def plot_image(image, scale):
|