Update app.py

app.py

@@ -2,12 +2,12 @@ import torch
 import numpy as np
 import pandas as pd
 import gradio as gr
-from PIL import Image
+from PIL import Image as PILIMAGE
 from transformers import CLIPProcessor, CLIPModel
 
 def find_similar(image):
 
-    image = Image.fromarray(image.astype('uint8'), 'RGB')
+    image = PILIMAGE.fromarray(image.astype('uint8'), 'RGB')
     device = "cuda" if torch.cuda.is_available() else "cpu"
     
     ## Define model
@@ -36,7 +36,7 @@ def find_similar(image):
     photo_id = photo_ids[idx]
     photo_data = photos[photos["photo_id"] == photo_id].iloc[0]
 
-    return Image(url=photo_data["photo_image_url"] + "?w=640")
+    return PILIMAGE(url=photo_data["photo_image_url"] + "?w=640")
         
 
 iface = gr.Interface(fn=find_similar, inputs=gr.inputs.Image(), outputs=gr.outputs.Image(type="pil")).launch()