Spaces:
Runtime error
Runtime error
Bill Psomas
commited on
Commit
·
2b1850a
1
Parent(s):
4f80be6
higher resolution choice
Browse files
app.py
CHANGED
|
@@ -45,7 +45,7 @@ msg = model.load_state_dict(state_dict, strict=True)
|
|
| 45 |
|
| 46 |
model.eval()
|
| 47 |
|
| 48 |
-
def get_attention_map(img, resolution):
|
| 49 |
input_size = int(resolution) * 14
|
| 50 |
data_transforms = transforms.Compose([
|
| 51 |
transforms.Resize((input_size, input_size), interpolation=3),
|
|
|
|
| 45 |
|
| 46 |
model.eval()
|
| 47 |
|
| 48 |
+
def get_attention_map(img, resolution=32):
|
| 49 |
input_size = int(resolution) * 14
|
| 50 |
data_transforms = transforms.Compose([
|
| 51 |
transforms.Resize((input_size, input_size), interpolation=3),
|