Files changed (1) hide show
  1. app.py +19 -2
app.py CHANGED
@@ -1,11 +1,28 @@
1
  import gradio as gr
2
  import os
3
  import json
 
4
  from fastai.vision.all import load_learner, PILImage
5
 
6
- gcp_json = json.loads(os.environ["gcp_cam"])
7
 
8
- learn = load_learner("cameroon_food.pkl")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
 
10
 
11
  def predict(img):
 
1
  import gradio as gr
2
  import os
3
  import json
4
+ from google.cloud import storage
5
  from fastai.vision.all import load_learner, PILImage
6
 
 
7
 
8
+ #Setting up GCP client
9
+ credentials_content = os.environ['gcp_cam']
10
+ with open('gcp_key.json', 'w') as f:
11
+ f.write(credentials_content)
12
+
13
+ os.environ['GOOGLE_APPLICATION_CREDENTIALS'] = 'gcp_key.json'
14
+ bucket_name = os.environ['gcp_bucket']
15
+ pkl_blob = 'paulinus/cameroon_food.pkl'
16
+ local_pkl = 'cameroon_food.pkl'
17
+
18
+ client = storage.Client()
19
+ bucket = client.bucket(bucket_name)
20
+ blob = bucket.blob(pkl_blob)
21
+ blob.download_to_filename(local_pkl)
22
+
23
+ #Load model
24
+ learn = load_learner(local_pkl)
25
+
26
 
27
 
28
  def predict(img):