Update src/streamlit_app.py

src/streamlit_app.py

@@ -20,50 +20,26 @@ from recommendation_utils import (
 #    recommend_with_nn, recommend_with_svd, load_encodings
 #)
 
-@st.cache_resource(show_spinner=False)
-def download_models_once():
-    DOWNLOAD_DIR = "/tmp"
-
-    files = {
-        #"encodings.pkl": "1EzpdpaopfUp-Tfc7YjxPVYQUwnU_BX5-",
-        #"config.json": "1ZcTrVR0QtS-5EL4amsTR_y9xITDA6zlJ",
-        #"model.weights.h5": "1VjjUx_7ulIVM-W1lqH-nDHI7HWfpcMQP",
-        "svd_model.pkl": "1fN2biQruVjJHHv2vX1g1hLuMeJoPqyFX",
-        "trainset.pkl": "1IDVVAQ57Xvf3HCAikbOSQgAigdHP7Ik7"
-    }
 
-    def gdrive_download(file_id, destination):
-        URL = "https://drive.google.com/uc?export=download"
-        session = requests.Session()
-        response = session.get(URL, params={'id': file_id}, stream=True)
-        token = None
-        for k, v in response.cookies.items():
-            if k.startswith('download_warning'):
-                token = v
-        if token:
-            response = session.get(URL, params={'id': file_id, 'confirm': token}, stream=True)
-        with open(destination, "wb") as f:
-            for chunk in response.iter_content(32768):
-                if chunk:
-                    f.write(chunk)
-
-    for filename, file_id in files.items():
-        full_path = os.path.join(DOWNLOAD_DIR, filename)
-        if not os.path.exists(full_path):
-            with st.spinner(f"Downloading {filename}..."):
-                gdrive_download(file_id, full_path)
-
-download_models_once()
+@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():
-    # nn_model = load_nn_model("/tmp/config.json", "/tmp/model.weights.h5")
-    svd_model = load_svd_model("/tmp/svd_model.pkl")
-    trainset = load_trainset("/tmp/trainset.pkl")
+    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")