Spaces:
Sleeping
Sleeping
Commit
·
0f98ba7
1
Parent(s):
8a8dac0
tried PLMS5
Browse files
app.py
CHANGED
@@ -49,7 +49,7 @@ def create_ds(img_paths):
|
|
49 |
]
|
50 |
return mn.data.Dataset(data_list, transform=mn.transforms.Compose(Ts_list))
|
51 |
|
52 |
-
def make_predictions(img_path, angles=None, cls_batch=None, rotate_to_standard=False, sampler="
|
53 |
|
54 |
global model
|
55 |
global BASELINE_NOISE
|
|
|
49 |
]
|
50 |
return mn.data.Dataset(data_list, transform=mn.transforms.Compose(Ts_list))
|
51 |
|
52 |
+
def make_predictions(img_path, angles=None, cls_batch=None, rotate_to_standard=False, sampler="PLMS5"):
|
53 |
|
54 |
global model
|
55 |
global BASELINE_NOISE
|