Update app.py

app.py

@@ -60,23 +60,14 @@ def resize_image_dimensions(
 
 
 def evf_sam_mask(image, prompt):
-    print(type(image))
-    filename=str(uuid.uuid4()) + ".jpg"
-    image.save(filename)
+    print(image)
     images = client.predict(
-      image_np=handle_file(filename),
+      image_np=handle_file(image),
       prompt=prompt, 
       api_name="/predict")
     print(images)
-    # Open the image
-    webp_image = Image.open(images[1])
-    # Convert to RGB mode if it's not already
-    if webp_image.mode != 'RGB':
-        webp_image = webp_image.convert('RGB')
-    # Create a new PIL Image object
-    pil_image = Image.new('RGB', webp_image.size)
-    pil_image.paste(webp_image)    
-
+    # Open the mask image
+    pil_image = Image.open(images[1])
     print(pil_image)
     print(type(pil_image))
 
@@ -138,28 +129,26 @@ with gr.Blocks() as demo:
     gr.Markdown(MARKDOWN)
     with gr.Row():
         with gr.Column():
-            input_image = gr.ImageEditor( #input_image_editor_component
+            input_image = gr.Image( 
                 label='Image',
-                type='pil',
-                sources=["upload", "webcam"],
+                type='filepath',
+                sources=["upload", "webcam", "clipboard"],
                 image_mode='RGB',
-                layers=False,
-                brush=gr.Brush(colors=["#FFFFFF"], color_mode="fixed"))
-
+            )
             with gr.Row():
                 with gr.Column():
                     input_text_component = gr.Text(
-                        label="Segment",
-                        show_label=False,
+                        label="Text-guided segmentation",
+                        show_label=True,
                         max_lines=1,
-                        placeholder="segmentation text",
+                        placeholder="Enter text for generating the segmentation mask",
                         container=False,
                     )
                     inpaint_text_component = gr.Text(
-                        label="Inpaint",
-                        show_label=False,
+                        label="Text-guided Inpainting",
+                        show_label=True,
                         max_lines=1,
-                        placeholder="Inpaint text",
+                        placeholder="Enter text to generate Inpainting",
                         container=False,
                     )
                 submit_button_component = gr.Button(value='Submit', variant='primary', scale=0)