Update app.py

app.py

@@ -9,7 +9,7 @@ import time
 from datetime import timedelta, datetime
 from branca.element import Figure
 
-from functions import decode_features, get_model, get_model1, get_model2
+from functions import decode_features
 
 
 def greet(total_pred_days):
@@ -33,18 +33,10 @@ def greet(total_pred_days):
     X = feature_view.get_batch_data(start_time=start_time)
     latest_date_unix = str(X.datetime.values[0])[:10]
     latest_date = time.ctime(int(latest_date_unix))
-    model = get_model(project=project,
-                      model_name="temp_model",
-                      evaluation_metric="f1_score",
-                      sort_metrics_by="max")
-    model1 = get_model1(project=project,
-                      model_name="tempmax_model",
-                      evaluation_metric="f1_score",
-                      sort_metrics_by="max") 
-    model2 = get_model2(project=project,
-                      model_name="tempmin_model",
-                      evaluation_metric="f1_score",
-                      sort_metrics_by="max")
+    model = get_model("temp_model", version=2)
+    model1 = get_model("tempmax_model", version=2)
+    model2 = get_model("tempmin_model", version=2)
+
     X = X.drop(columns=["datetime"]).fillna(0)
 
     preds = model.predict(X)