Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import torch
+from PIL import Image
 from diffusers import StableDiffusionInpaintPipeline
 
 # Load the Stable Diffusion inpainting model
@@ -14,6 +15,10 @@ def load_inpainting_model():
 
 # Function to edit the clothing image based on prompt
 def edit_clothing_image(prompt, image):
+    # Ensure the image is in PIL format
+    if isinstance(image, np.ndarray):
+        image = Image.fromarray(image)
+    
     # Load the model
     model = load_inpainting_model()