5m4ck3r commited on
Commit
322cdde
1 Parent(s): 0a275ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
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(base64_image: str, text: str):
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
- [gradio.Image(type="filepath"), gradio.Text()],
105
- gradio.Image(type="pil")
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)