Spaces:
Runtime error
Runtime error
Daniel Bustamante Ospina
commited on
Commit
•
cae220d
1
Parent(s):
608e54d
New bidirectional model
Browse files- app.py +1 -1
- model_scripted.pt_enc +0 -0
- 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
|
|