Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,11 +4,15 @@ import numpy as np
|
|
| 4 |
from PIL import Image
|
| 5 |
import requests
|
| 6 |
from io import BytesIO
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
|
| 8 |
# Load model
|
| 9 |
@st.cache_resource
|
| 10 |
def load_model():
|
| 11 |
-
return tf.keras.models.load_model('fish_classification_model.h5')
|
| 12 |
|
| 13 |
model = load_model()
|
| 14 |
|
|
@@ -30,9 +34,6 @@ def get_fish_emoji(fish_name):
|
|
| 30 |
}
|
| 31 |
return emoji_dict.get(fish_name, '🐠')
|
| 32 |
|
| 33 |
-
# Streamlit app
|
| 34 |
-
st.set_page_config(page_title="Fish Species Classifier", page_icon="🐠", layout="wide")
|
| 35 |
-
|
| 36 |
# Add a background image
|
| 37 |
background_image = """
|
| 38 |
<style>
|
|
|
|
| 4 |
from PIL import Image
|
| 5 |
import requests
|
| 6 |
from io import BytesIO
|
| 7 |
+
import pandas as pd
|
| 8 |
+
|
| 9 |
+
# Move st.set_page_config() to the top
|
| 10 |
+
st.set_page_config(page_title="Fish Species Classifier", page_icon="🐠", layout="wide")
|
| 11 |
|
| 12 |
# Load model
|
| 13 |
@st.cache_resource
|
| 14 |
def load_model():
|
| 15 |
+
return tf.keras.models.load_model('fish_classification_model.h5')
|
| 16 |
|
| 17 |
model = load_model()
|
| 18 |
|
|
|
|
| 34 |
}
|
| 35 |
return emoji_dict.get(fish_name, '🐠')
|
| 36 |
|
|
|
|
|
|
|
|
|
|
| 37 |
# Add a background image
|
| 38 |
background_image = """
|
| 39 |
<style>
|