AkinyemiAra commited on
Commit
afcdacc
·
verified ·
1 Parent(s): 7f4fed9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -39,15 +39,13 @@ def focus_on_subject(image: Image.Image) -> Image.Image:
39
 
40
  return cropped.convert("RGB")
41
 
42
- def ImgEmbed(image: Image.Image):
43
  """
44
  Preprocess image, generate normalized embedding, and return both embedding and processed image.
45
-
46
  Args:
47
  image (PIL.Image.Image): Input image.
48
-
49
  Returns:
50
- Tuple: (embedding list, processed image)
51
  """
52
  focused_image = focus_on_subject(image)
53
  inputs = processor(focused_image, return_tensors="pt")
 
39
 
40
  return cropped.convert("RGB")
41
 
42
+ def ImgEmbed(image: Image.Image)-> List[float]:
43
  """
44
  Preprocess image, generate normalized embedding, and return both embedding and processed image.
 
45
  Args:
46
  image (PIL.Image.Image): Input image.
 
47
  Returns:
48
+ List[float]: Embedding vector.
49
  """
50
  focused_image = focus_on_subject(image)
51
  inputs = processor(focused_image, return_tensors="pt")