Update app.py

app.py

@@ -2,39 +2,29 @@ import os
 import requests
 import gradio as gr
 
-from utils import get_token
-
 url_caption = os.environ["CAPTION_NODE"]
 url_vqa = os.environ["VQA_NODE"]
 
 
 def image_caption(file_path):
-    token = get_token()
-
-    files = {"file": open(file_path, "rb")}
-    headers = {"X-Auth-Token": token}
+    files = {"picture": open(file_path, "rb")}
     resp = requests.post(url_caption,
                          files=files,
-                         headers=headers,
                          verify=False)
     resp = resp.json()
-    desc = resp["inference_result"]["instances"]["image"][0]
+    desc = resp["data"]["desc"]
     return desc
 
 
 def vqa(file_path, question):
-    token = get_token()
-
-    files = {"file": open(file_path, "rb")}
+    files = {"picture": open(file_path, "rb")}
     question = {"question": question}
-    headers = {"X-Auth-Token": token}
     resp = requests.post(url_vqa,
                          files=files,
                          data=question,
-                         headers=headers,
                          verify=False)
     resp = resp.json()
-    ans = resp["inference_result"]["instances"]
+    ans = resp["data"]["answer"]
     return ans
 
 
@@ -63,7 +53,6 @@ examples_vqa = [
 
 css = """
 .gradio-container {background-image: url('file=./background.jpg'); background-size:cover; background-repeat: no-repeat;}
-
 #infer {
     background: linear-gradient(to bottom right, #FFD8B4, #FFB066);
     border: 1px solid #ffd8b4;