Update app.py

app.py

@@ -2,27 +2,26 @@ import streamlit as st
 import pandas as pd
 import json
 import joblib
-import requests
-from io import BytesIO
-from zipfile import ZipFile
+from huggingface_hub import hf_hub_download
 
 # Display the version of huggingface_hub
 import huggingface_hub
 st.write(f"Hugging Face Hub version: {huggingface_hub.__version__}")
 
-# Function to download models from Hugging Face using requests
+# Function to download models from Hugging Face using huggingface_hub
 def download_model_from_huggingface(repo_id, filename):
     """Downloads a model file from Hugging Face repository."""
-    url = f"https://huggingface.co/{repo_id}/resolve/main/{filename}"
-    response = requests.get(url)
-    if response.status_code == 200:
-        return BytesIO(response.content)
-    else:
-        raise Exception(f"Failed to download {filename}")
+    try:
+        # Using huggingface_hub API to download model directly
+        model_file = hf_hub_download(repo_id, filename)
+        return model_file
+    except Exception as e:
+        st.error(f"Error downloading model {filename}: {e}")
+        return None
 
 # Function to load models safely
 def load_model(model_file):
-    """Loads a model from a BytesIO object."""
+    """Loads a model from a file."""
     try:
         # Here we're assuming the model is in a `.joblib` file format
         return joblib.load(model_file)