Update main.py
Browse files
main.py
CHANGED
@@ -1,15 +1,14 @@
|
|
1 |
from flask import Flask, request, jsonify
|
2 |
-
import asyncio
|
3 |
import os
|
4 |
-
from hiou import hugging
|
5 |
import json
|
6 |
import io
|
7 |
|
8 |
-
def get_answer(image:
|
9 |
-
filename = image
|
10 |
spid = os.getenv('SPID')
|
11 |
hug = hugging(spid)
|
12 |
-
filelink = hug.upload(
|
13 |
image.seek(0, 2)
|
14 |
file_size = image.tell()
|
15 |
image.seek(0)
|
@@ -23,16 +22,19 @@ def get_answer(image: str, question: str):
|
|
23 |
"path": filelink.split("=", 1)[1],
|
24 |
"size": file_size,
|
25 |
"url": filelink
|
26 |
-
},
|
27 |
-
question,
|
28 |
-
{
|
29 |
-
"tab_index": 0
|
30 |
}
|
31 |
]
|
32 |
datas = os.getenv('DATA')
|
33 |
jsnd = json.loads(datas)
|
34 |
data.append(jsnd)
|
35 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
36 |
hug.filnal_setup(data, 2, 15)
|
37 |
hug.start()
|
38 |
return hug.output.get("data")[0]
|
@@ -59,3 +61,4 @@ def async_task():
|
|
59 |
def home():
|
60 |
return jsonify({"message": "Welcome to my Flask API!"})
|
61 |
|
|
|
|
1 |
from flask import Flask, request, jsonify
|
|
|
2 |
import os
|
3 |
+
from hiou import hugging
|
4 |
import json
|
5 |
import io
|
6 |
|
7 |
+
def get_answer(image: io.BytesIO, question: str):
|
8 |
+
filename = image.name
|
9 |
spid = os.getenv('SPID')
|
10 |
hug = hugging(spid)
|
11 |
+
filelink = hug.upload(image)
|
12 |
image.seek(0, 2)
|
13 |
file_size = image.tell()
|
14 |
image.seek(0)
|
|
|
22 |
"path": filelink.split("=", 1)[1],
|
23 |
"size": file_size,
|
24 |
"url": filelink
|
|
|
|
|
|
|
|
|
25 |
}
|
26 |
]
|
27 |
datas = os.getenv('DATA')
|
28 |
jsnd = json.loads(datas)
|
29 |
data.append(jsnd)
|
30 |
+
data.append(
|
31 |
+
question
|
32 |
+
)
|
33 |
+
data.append(
|
34 |
+
{
|
35 |
+
"tab_index": 0
|
36 |
+
}
|
37 |
+
)
|
38 |
hug.filnal_setup(data, 2, 15)
|
39 |
hug.start()
|
40 |
return hug.output.get("data")[0]
|
|
|
61 |
def home():
|
62 |
return jsonify({"message": "Welcome to my Flask API!"})
|
63 |
|
64 |
+
app.run()
|