Update src/streamlit_app.py

src/streamlit_app.py

@@ -8,9 +8,7 @@ import requests
 import difflib
 import pickle
 from keras.models import load_model
-import os
 import requests
-import os
 
 os.environ["SURPRISE_DATA_FOLDER"] = "/tmp/.surprise_data"
 
@@ -24,26 +22,6 @@ from recommendation_utils import (
 #    recommend_with_nn, recommend_with_svd, load_encodings
 #)
 
-
-@st.cache_resource
-def load_remote_pickle(url):
-    response = requests.get(url)
-    response.raise_for_status()
-    return pickle.loads(response.content)
-
-@st.cache_resource
-def load_models():
-    SVD_URL = "https://huggingface.co/lenawilli/App_models_Py/resolve/main/svd_model.pkl"
-    TRAINSET_URL = "https://huggingface.co/lenawilli/App_models_Py/resolve/main/trainset.pkl"
-
-    svd_model = load_remote_pickle(SVD_URL)
-    trainset = load_remote_pickle(TRAINSET_URL)
-
-    return svd_model, trainset
-
-svd_model, trainset = load_models()
-
-
 # encodings = load_encodings("/tmp/encodings.pkl")
 
 st.set_page_config(layout="wide")
@@ -194,15 +172,6 @@ st.markdown("""
     </div>
 """, unsafe_allow_html=True)
 
-# Modell-Auswahl Dropdown
-model_choice = st.radio(
-    "Choose Recommendation Model:",
-    options=["Neural Network", "SVD"],
-    index=0,
-    horizontal=True,
-    key="model_selection"
-)
-
 query_params = st.query_params
 page = query_params.get("rateflow")
 search_query = query_params.get("search")
@@ -458,6 +427,33 @@ elif search_query:
 else:
     st.title("Welcome to Movie Recommender")
 
+    # Modell-Auswahl Dropdown
+    model_choice = st.radio(
+        "Choose Recommendation Model:",
+        options=["Neural Network", "SVD"],
+        index=0,
+        horizontal=True,
+        key="model_selection"
+    )
+    
+    @st.cache_resource
+    def load_remote_pickle(url):
+        response = requests.get(url)
+        response.raise_for_status()
+        return pickle.loads(response.content)
+    
+    @st.cache_resource
+    def load_models():
+        SVD_URL = "https://huggingface.co/lenawilli/App_models_Py/resolve/main/svd_model.pkl"
+        TRAINSET_URL = "https://huggingface.co/lenawilli/App_models_Py/resolve/main/trainset.pkl"
+    
+        svd_model = load_remote_pickle(SVD_URL)
+        trainset = load_remote_pickle(TRAINSET_URL)
+    
+        return svd_model, trainset
+    
+    svd_model, trainset = load_models()
+
     if not all_ratings_data:
         st.info("No ratings available yet. Start rating some movies!")
     else: