Update app.py

app.py

@@ -1,5 +1,3 @@
-from IPython import get_ipython
-from IPython.display import display
 from transformers import SegformerImageProcessor, AutoModelForSemanticSegmentation
 from PIL import Image
 import requests
@@ -10,8 +8,7 @@ from torchvision import transforms
 from transformers import SamModel, SamProcessor
 from diffusers import AutoPipelineForInpainting
 from diffusers.utils import load_image, make_image_grid
-from google.colab import drive
-drive.mount('/content/drive')
+
 
 
 def modify_image(image_url, prompt, mask_id=4):
@@ -62,7 +59,7 @@ def gradio_wrapper(image, prompt, choice):
 demo = gr.Interface(
     fn=gradio_wrapper,
     inputs=[
-        gr.Image(type="filepath"),  # Change gr.inputs.Image to gr.Image
+        gr.Image(type="filepath"), 
         gr.Textbox(label="Prompt"),
         gr.Radio(["4", "5", "6"], label="Mask ID")
     ],