Update app.py

app.py

@@ -10,6 +10,7 @@ import os
 import gradio as gr
 import time
 from torchvision import transforms
+from PIL import Image
 
 img_mean = [0.485, 0.456, 0.406]
 img_std = [0.229, 0.224, 0.225]
@@ -23,6 +24,7 @@ def inference_mask1(
             query_img,
             *prompt,
               ):
+    query_img = Image(query_img)
     query_img_np = np.asarray(query_img)
     query_img = transformation(query_img)
     shape = query_img_np.shape