RobotJelly commited on
Commit
de0179f
1 Parent(s): 3c7ac16
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -72,7 +72,7 @@ def image_search(search_text, search_image, option):
72
  return show_output_image(matched_images)
73
  elif option == "Image-To-Image":
74
  # Input Image for Search
75
- search_image = PILIMAGE.fromarray(search_image.astype('uint8'), 'RGB')
76
 
77
  with torch.no_grad():
78
  processed_image = processor(text=None, images=search_image, return_tensors="pt", padding=True)["pixel_values"]
 
72
  return show_output_image(matched_images)
73
  elif option == "Image-To-Image":
74
  # Input Image for Search
75
+ search_image = Image.fromarray(search_image.astype('uint8'), 'RGB')
76
 
77
  with torch.no_grad():
78
  processed_image = processor(text=None, images=search_image, return_tensors="pt", padding=True)["pixel_values"]