Added secrets

app.py

@@ -8,6 +8,7 @@ import  io
 import json
 from numba import jit
 import matplotlib.pyplot as plt
+import os 
 
 
 examples =  ["examples/0002_01_00_01_55.jpg",
@@ -21,10 +22,10 @@ examples =  ["examples/0002_01_00_01_55.jpg",
 
 
 async def spoof_trigger(b64):
-    url = "https://spoofapi1.azurewebsites.net/api/spoofvisualize"
+    url = os.getenv('url') 
     payload = {"img": b64}
     headers = {
-    'x-functions-key': 'wGw3zXXPlLCez-VrcSs9RTahE4gLC674pf7Fp6Au2kUHAzFuNnZZMw==',
+    'x-functions-key': os.getenv('token'),
     'Content-Type': 'text/plain'
     }
 
@@ -57,16 +58,16 @@ async def predict_image(img):
     # print(json.loads(res))
     spoof_res = json.loads(res)['spoof_res']
     annotated_image  = json.loads(res)['annotated_image']
-    conf_score = float( json.loads(spoof_res)['confidence_score'])
+    conf_score = float( json.loads(spoof_res)['confidence_score2'])
 
     # img_base64 to  plot
     img = Image.open(io.BytesIO(base64.b64decode(annotated_image)))
     # img save 
-    img.save("cache/img.jpg")
+    # img.save("cache/img.jpg")
 
-    confidences = {'Real': 1-conf_score, 'Fake': conf_score}
+    confidences = {'Real': conf_score, 'Fake': 1-conf_score}
 
-    return (confidences,annotated_image)
+    return (confidences,img)
 
 
 with gr.Blocks(title="Spoof-Demo", css="#custom_header {min-height: 3rem; text-align:  center} #custom_title {min-height: 3rem; text-align: center}") as demo :