yonadab commited on
Commit
cef5090
·
verified ·
1 Parent(s): d871817

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -7,6 +7,7 @@ processor = AutoImageProcessor.from_pretrained("facebook/dinov2-base")
7
  model = AutoModel.from_pretrained("facebook/dinov2-base")
8
 
9
  def get_embedding(image):
 
10
  inputs = processor(images=image, return_tensors="pt")
11
  with torch.no_grad():
12
  embeddings = model(**inputs).last_hidden_state[:, 0] # CLS token
 
7
  model = AutoModel.from_pretrained("facebook/dinov2-base")
8
 
9
  def get_embedding(image):
10
+ print("Tipo de imagen recibido:", type(image))
11
  inputs = processor(images=image, return_tensors="pt")
12
  with torch.no_grad():
13
  embeddings = model(**inputs).last_hidden_state[:, 0] # CLS token