ColinceTatsa commited on
Commit
04c3877
·
verified ·
1 Parent(s): afe3cf8

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -137,7 +137,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("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.")
 
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.")