Update app.py

app.py

@@ -3,6 +3,7 @@ from PIL import Image
 import numpy as np
 from transformers import SamModel, SamProcessor
 from diffusers import AutoPipelineForInpainting
+from diffusers.models.autoencoders.vq_model import VQEncoderOutput, VQModel
 import torch
 
 # Force the model to use CPU
@@ -58,9 +59,10 @@ def gradio_interface(image, points):
 iface = gr.Interface(
     fn=gradio_interface,
     inputs=[
-        gr.Image(image_mode='RGB', image_shape=(512, 512)), 
+        gr.Image(image_mode='RGB'),  # Removed image_shape parameter
         "text"  # To allow users to enter points manually or through another input method
     ],
     outputs=["image", "image"]
 )
+
 iface.launch(share=True)