Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -89,9 +89,7 @@ def process_image(image):
|
|
89 |
processed_image.putdata(final_data)
|
90 |
return processed_image
|
91 |
|
92 |
-
def processAndGetMask(
|
93 |
-
image_bytes = base64.b64decode(base64_image.split(',')[1])
|
94 |
-
image = Image.open(BytesIO(image_bytes))
|
95 |
datas = getImageDetails(image)
|
96 |
labs = list(datas.keys())
|
97 |
selector = processSentence(text, labs)
|
@@ -101,7 +99,7 @@ def processAndGetMask(base64_image: str, text: str):
|
|
101 |
|
102 |
gr = gradio.Interface(
|
103 |
processAndGetMask,
|
104 |
-
[
|
105 |
-
|
106 |
)
|
107 |
gr.launch(share=True)
|
|
|
89 |
processed_image.putdata(final_data)
|
90 |
return processed_image
|
91 |
|
92 |
+
def processAndGetMask(image: str, text: str):
|
|
|
|
|
93 |
datas = getImageDetails(image)
|
94 |
labs = list(datas.keys())
|
95 |
selector = processSentence(text, labs)
|
|
|
99 |
|
100 |
gr = gradio.Interface(
|
101 |
processAndGetMask,
|
102 |
+
[gr.inputs.Image(label="image", type="pil"), "text"],
|
103 |
+
"image"
|
104 |
)
|
105 |
gr.launch(share=True)
|