Update app.py

app.py

@@ -86,7 +86,9 @@ def process_image(image):
     processed_image.putdata(final_data)
     return processed_image
 
-def processAndGetMask(image: str, text: str):
+def processAndGetMask(base64_image: str, text: str):
+    image_bytes = base64.b64decode(base64_image.split(',')[1])
+    image = Image.open(BytesIO(image_bytes))
     datas = getImageDetails(image)
     labs = list(datas.keys())
     selector = processSentence(text, labs)
@@ -96,7 +98,7 @@ def processAndGetMask(image: str, text: str):
 
 gr = gradio.Interface(
     processAndGetMask,
-    [gradio.Image(type="pil"), gradio.Text()],
+    [gradio.Image(), gradio.Text()],
     gradio.Image(type="pil")
 )
 gr.launch(share=True)