Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +5 -4
src/streamlit_app.py
CHANGED
|
@@ -18,6 +18,7 @@ st.set_page_config(
|
|
| 18 |
)
|
| 19 |
|
| 20 |
script = Path(__file__)
|
|
|
|
| 21 |
|
| 22 |
|
| 23 |
# --- Définition des modèles et fonctions (partie non visible dans l'UI) ---
|
|
@@ -37,7 +38,7 @@ class GRUModel(nn.Module):
|
|
| 37 |
|
| 38 |
# === Fonctions de chargement et de traitement (avec cache pour la performance) ===
|
| 39 |
@st.cache_data
|
| 40 |
-
def load_data(csv_path=
|
| 41 |
"""Charge les données depuis le fichier CSV et les formate correctement."""
|
| 42 |
try:
|
| 43 |
df = pd.read_csv(csv_path)
|
|
@@ -98,7 +99,7 @@ def page_accueil():
|
|
| 98 |
col1, col2 = st.columns([1, 3])
|
| 99 |
with col1:
|
| 100 |
try:
|
| 101 |
-
logo_keyce = Image.open(
|
| 102 |
st.image(logo_keyce, width=150)
|
| 103 |
except FileNotFoundError:
|
| 104 |
st.warning("Logo Keyce Keyce_Logo.jpg non trouvé.")
|
|
@@ -137,7 +138,7 @@ def page_prediction():
|
|
| 137 |
st.header("Étape 2 : Chargement du Modèle GRU")
|
| 138 |
with st.spinner("Chargement du modèle pré-entraîné..."):
|
| 139 |
try:
|
| 140 |
-
gru_model = load_gru_model(
|
| 141 |
st.success("Modèle GRU chargé avec succès !")
|
| 142 |
except FileNotFoundError:
|
| 143 |
st.error("Erreur : Le fichier 'model_gru.pth' est introuvable.")
|
|
@@ -198,7 +199,7 @@ def page_prediction():
|
|
| 198 |
|
| 199 |
st.sidebar.header("Navigation")
|
| 200 |
try:
|
| 201 |
-
logo_theme = Image.open(
|
| 202 |
st.sidebar.image(logo_theme, use_container_width=True)
|
| 203 |
except FileNotFoundError:
|
| 204 |
st.sidebar.warning("Logo thème Theme_Logo.jpg non trouvé.")
|
|
|
|
| 18 |
)
|
| 19 |
|
| 20 |
script = Path(__file__)
|
| 21 |
+
script_dir = script_path.parent
|
| 22 |
|
| 23 |
|
| 24 |
# --- Définition des modèles et fonctions (partie non visible dans l'UI) ---
|
|
|
|
| 38 |
|
| 39 |
# === Fonctions de chargement et de traitement (avec cache pour la performance) ===
|
| 40 |
@st.cache_data
|
| 41 |
+
def load_data(csv_path= script_dir/"action_amd.csv"):
|
| 42 |
"""Charge les données depuis le fichier CSV et les formate correctement."""
|
| 43 |
try:
|
| 44 |
df = pd.read_csv(csv_path)
|
|
|
|
| 99 |
col1, col2 = st.columns([1, 3])
|
| 100 |
with col1:
|
| 101 |
try:
|
| 102 |
+
logo_keyce = Image.open(script_dir/"Keyce_Logo.jpg")
|
| 103 |
st.image(logo_keyce, width=150)
|
| 104 |
except FileNotFoundError:
|
| 105 |
st.warning("Logo Keyce Keyce_Logo.jpg non trouvé.")
|
|
|
|
| 138 |
st.header("Étape 2 : Chargement du Modèle GRU")
|
| 139 |
with st.spinner("Chargement du modèle pré-entraîné..."):
|
| 140 |
try:
|
| 141 |
+
gru_model = load_gru_model(script_dir/"model_gru.pth", GRUModel)
|
| 142 |
st.success("Modèle GRU chargé avec succès !")
|
| 143 |
except FileNotFoundError:
|
| 144 |
st.error("Erreur : Le fichier 'model_gru.pth' est introuvable.")
|
|
|
|
| 199 |
|
| 200 |
st.sidebar.header("Navigation")
|
| 201 |
try:
|
| 202 |
+
logo_theme = Image.open(script_dir/"Theme_Logo.jpg")
|
| 203 |
st.sidebar.image(logo_theme, use_container_width=True)
|
| 204 |
except FileNotFoundError:
|
| 205 |
st.sidebar.warning("Logo thème Theme_Logo.jpg non trouvé.")
|