Spaces:

RobotJelly
/

Text_Or_Image-To-Image_Search

Build error

RobotJelly commited on Nov 23, 2021

Commit

de0179f

1 Parent(s): 3c7ac16

app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -72,7 +72,7 @@ def image_search(search_text, search_image, option):
     return show_output_image(matched_images)
   elif option == "Image-To-Image":
       # Input Image for Search
-      search_image = PILIMAGE.fromarray(search_image.astype('uint8'), 'RGB')
       with torch.no_grad():
          processed_image = processor(text=None, images=search_image, return_tensors="pt", padding=True)["pixel_values"]

     return show_output_image(matched_images)
   elif option == "Image-To-Image":
       # Input Image for Search
+      search_image = Image.fromarray(search_image.astype('uint8'), 'RGB')
       with torch.no_grad():
          processed_image = processor(text=None, images=search_image, return_tensors="pt", padding=True)["pixel_values"]