Hugging Face's logo

Spaces:
dbuos
/
dog_recognition
Runtime error

App Files Community
Daniel Bustamante Ospina commited on
Commit
cae220d
·
1 Parent(s): 608e54d

New bidirectional model

Browse files
Files changed (3) hide show
  1. app.py +1 -1
  2. model_scripted.pt_enc +0 -0
  3. model_scripted_old0.pt_enc +0 -0
app.py CHANGED
@@ -29,7 +29,7 @@ def predict_diff(img_a, img_b):
29
  global model_cls, feat_extractor, processor
30
  x = processor(img_a).unsqueeze(dim=0).to(device), processor(img_b).unsqueeze(dim=0).to(device)
31
  a, b = feat_extractor(x)
32
- proba = torch.sigmoid(model_cls(a, b)).item()
33
  score_str = "{:.2f}".format(round(proba) * proba + round(1 - proba) * (1 - proba))
34
  base_name = f"{str(uuid.uuid4()).replace('-', '')}-{score_str}"
35
  save_image_pairs(img_a, img_b, proba, base_name)
 
29
  global model_cls, feat_extractor, processor
30
  x = processor(img_a).unsqueeze(dim=0).to(device), processor(img_b).unsqueeze(dim=0).to(device)
31
  a, b = feat_extractor(x)
32
+ proba = torch.sigmoid(model_cls((a, b))).item()
33
  score_str = "{:.2f}".format(round(proba) * proba + round(1 - proba) * (1 - proba))
34
  base_name = f"{str(uuid.uuid4()).replace('-', '')}-{score_str}"
35
  save_image_pairs(img_a, img_b, proba, base_name)
model_scripted.pt_enc CHANGED
The diff for this file is too large to render. See raw diff
 
model_scripted_old0.pt_enc ADDED
The diff for this file is too large to render. See raw diff