Update app.py

app.py

@@ -5,10 +5,10 @@ from transformers import SamModel, SamProcessor
 from diffusers import AutoPipelineForInpainting
 import torch
 
-# Force the model to use CPU regardless of the availability of CUDA
+# Force the model to use CPU
 device = "cpu"
 
-# Update model loading and processing to work on CPU
+# Model and Processor setup
 model_name = "facebook/sam-vit-huge"
 model = SamModel.from_pretrained(model_name).to(device)
 processor = SamProcessor.from_pretrained(model_name)
@@ -48,7 +48,7 @@ def inpaint(raw_image, input_mask, prompt, negative_prompt=None, seed=74294536,
     ).images[0]
     return image
 
-# Gradio Interface with Click Events
+# Updated Gradio interface
 def gradio_interface(image, points):
     raw_image = Image.fromarray(image).convert("RGB").resize((512, 512))
     mask = get_processed_inputs(raw_image, points)
@@ -57,7 +57,10 @@ def gradio_interface(image, points):
 
 iface = gr.Interface(
     fn=gradio_interface,
-    inputs=["image", gr.Image(shape=(512, 512), image_mode='RGB', source="canvas", tool="sketch")],
+    inputs=[
+        gr.Image(image_mode='RGB', image_shape=(512, 512)), 
+        "text"  # To allow users to enter points manually or through another input method
+    ],
     outputs=["image", "image"]
 )
 iface.launch(share=True)