pricefix

app.py

@@ -13,6 +13,7 @@ from geopy.geocoders import Nominatim
 from datetime import date
 from datetime import timedelta
 from autogluon.tabular import TabularPredictor
+import shutil
 
 # Login to hopsworks and get the feature store
 
@@ -54,14 +55,15 @@ def downloadAutogluonModel():
     temp = mr.get_model("ag_model_20230109", version=3)
     temp_ag_folder_path = temp.download()
     print(temp_ag_folder_path)
+    moveFolder(temp_ag_folder_path)
 
-    ag_model = TabularPredictor.load(temp_ag_folder_path + '/ag_model_20230109') # '/ag_model_20230109'
+    ag_model = TabularPredictor.load("AutogluonModels/ag_model_20230109") # '/ag_model_20230109'
 
     return ag_model
 
+
 def moveFolder(temp_ag_folder_path):
     # Move Autogluon model folder to the correct folder
-    import shutil
     original = temp_ag_folder_path
     target = "AutogluonModels/"
     shutil.move(original, target)

requirements.txt

@@ -8,4 +8,5 @@ scikit-learn
 tqdm
 xgboost==1.7.2
 geopy
-autogluon
+autogluon
+shutil