minor import change

Dhanush123 · Feb 27, 2020 · 3d8226d · 3d8226d
1 parent 7c32a10
commit 3d8226d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 22 deletions.
diff --git a/cloudbuild.yaml b/cloudbuild.yaml
diff --git a/test1.py b/test1.py
@@ -1,8 +1,9 @@
 import os
 
 from flask import Flask, jsonify
-from gcp_utils.googlebucket import gcs_download_file
+from shaktiutils.gcp_utils.googlebucket import gcs_download_file
 from joblib import load
+from dotenv import load_dotenv
 
 app = Flask(__name__)
 
@@ -12,6 +13,7 @@
 @app.before_request
 def load_resources():
     '''Flask template is currently only for scikit-learn models'''
+    load_dotenv()
     if not model:
         model_path = gcs_download_file(os.environ.get("MODEL_PATH"))
         model = load(model_path)