1sozidatel1 commited on
Commit
8e365f3
·
verified ·
1 Parent(s): 5490850

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -3,9 +3,13 @@ import tensorflow as tf
3
  import numpy as np
4
  import pandas as pd
5
  from datasets import load_dataset
 
 
 
 
6
 
7
  # Загрузка модели
8
- model = tf.keras.models.load_model('model/price_model.h5')
9
 
10
  # Карты местоположений и районов
11
  location_mapping = {1: 'Астрахань', 2: 'Волгоград', 3: 'Краснодар', 4: 'Ростов-на-Дону', 5: 'Майкоп', 6: 'Элиста'}
 
3
  import numpy as np
4
  import pandas as pd
5
  from datasets import load_dataset
6
+ from huggingface_hub import hf_hub_download
7
+
8
+ # Загрузка файла модели из Hugging Face Hub
9
+ model_path = hf_hub_download(repo_id="1sozidatel1/PricePredictorAI_Model", filename="price_model.h5")
10
 
11
  # Загрузка модели
12
+ model = tf.keras.models.load_model(model_path)
13
 
14
  # Карты местоположений и районов
15
  location_mapping = {1: 'Астрахань', 2: 'Волгоград', 3: 'Краснодар', 4: 'Ростов-на-Дону', 5: 'Майкоп', 6: 'Элиста'}