Spaces:
Runtime error
Runtime error
linjieccc
commited on
Commit
•
05f882f
1
Parent(s):
6542ea8
update
Browse files
app.py
CHANGED
@@ -221,6 +221,12 @@ def process_upload(file):
|
|
221 |
)
|
222 |
|
223 |
|
|
|
|
|
|
|
|
|
|
|
|
|
224 |
def process_prompt(prompt, document, lang="ch"):
|
225 |
if not prompt:
|
226 |
prompt = "What is the total actual and/or obligated expenses of ECG Center?"
|
@@ -231,7 +237,7 @@ def process_prompt(prompt, document, lang="ch"):
|
|
231 |
|
232 |
url = f"https://aip.baidubce.com/rpc/2.0/nlp-itec/poc/docprompt?access_token={access_token}"
|
233 |
|
234 |
-
r = requests.post(url, json={"doc": document, "prompt": [prompt], "lang": lang})
|
235 |
response = r.json()
|
236 |
|
237 |
predictions = response['result']
|
|
|
221 |
)
|
222 |
|
223 |
|
224 |
+
def img_base64(img_path):
|
225 |
+
with open(img_path, "rb") as f:
|
226 |
+
base64_str = base64.b64encode(f.read())
|
227 |
+
return base64_str
|
228 |
+
|
229 |
+
|
230 |
def process_prompt(prompt, document, lang="ch"):
|
231 |
if not prompt:
|
232 |
prompt = "What is the total actual and/or obligated expenses of ECG Center?"
|
|
|
237 |
|
238 |
url = f"https://aip.baidubce.com/rpc/2.0/nlp-itec/poc/docprompt?access_token={access_token}"
|
239 |
|
240 |
+
r = requests.post(url, json={"doc": img_base64(document), "prompt": [prompt], "lang": lang})
|
241 |
response = r.json()
|
242 |
|
243 |
predictions = response['result']
|