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)
|