Spaces:
Sleeping
Sleeping
bug fix
Browse files
app.py
CHANGED
@@ -33,7 +33,7 @@ min_len = 1e10
|
|
33 |
for d in domains:
|
34 |
with open(os.path.join(feat_dir, f'src_{d}_{shots}.pkl'), 'rb') as fp:
|
35 |
dst_data_dict[d] = pickle.load(fp)
|
36 |
-
min_len = min(min_len,(dst_data_dict[d][0]))
|
37 |
|
38 |
def query(query_index, query_domain):
|
39 |
dst_data = dst_data_dict[query_domain]
|
|
|
33 |
for d in domains:
|
34 |
with open(os.path.join(feat_dir, f'src_{d}_{shots}.pkl'), 'rb') as fp:
|
35 |
dst_data_dict[d] = pickle.load(fp)
|
36 |
+
min_len = min(min_len, len(dst_data_dict[d][0]))
|
37 |
|
38 |
def query(query_index, query_domain):
|
39 |
dst_data = dst_data_dict[query_domain]
|