Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -9,6 +9,7 @@ from sklearn.preprocessing import LabelEncoder
|
|
9 |
import requests # Add this at the top with other imports
|
10 |
from io import BytesIO
|
11 |
import gdown
|
|
|
12 |
|
13 |
# --- Set page configuration ---
|
14 |
st.set_page_config(
|
@@ -218,8 +219,12 @@ def load_model_and_encodings():
|
|
218 |
try:
|
219 |
# Show loading message
|
220 |
with st.spinner('Loading model...'):
|
221 |
-
|
222 |
-
|
|
|
|
|
|
|
|
|
223 |
|
224 |
# Load data for encodings
|
225 |
original_data = load_datasets()
|
|
|
9 |
import requests # Add this at the top with other imports
|
10 |
from io import BytesIO
|
11 |
import gdown
|
12 |
+
from huggingface_hub import hf_hub_download
|
13 |
|
14 |
# --- Set page configuration ---
|
15 |
st.set_page_config(
|
|
|
219 |
try:
|
220 |
# Show loading message
|
221 |
with st.spinner('Loading model...'):
|
222 |
+
try:
|
223 |
+
model_content = hf_hub_download(repo_id="EdBoy2202/car_prediction_model", filename="car_price_modelv3.pkl")
|
224 |
+
model = joblib.load(BytesIO(model_content))
|
225 |
+
except:
|
226 |
+
model_content = download_file_from_google_drive(model_file_id)
|
227 |
+
model = joblib.load(BytesIO(model_content))
|
228 |
|
229 |
# Load data for encodings
|
230 |
original_data = load_datasets()
|