ColinceTatsa commited on
Commit
c4c84cb
·
verified ·
1 Parent(s): 38bfd19

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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= script/"action_amd.csv"):
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(script/"Keyce_Logo.jpg")
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(script/"model_gru.pth", GRUModel)
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(script/"Theme_Logo.jpg")
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é.")