Spaces:
Running
Running
dev(narugo): add konachan
Browse files
app.py
CHANGED
@@ -56,7 +56,7 @@ def _get_from_ids(site_name: str, ids: List[int]) -> Dict[int, Image.Image]:
|
|
56 |
def _get_from_raw_ids(ids: List[str]) -> Dict[str, Image.Image]:
|
57 |
_sites = defaultdict(list)
|
58 |
for id_ in ids:
|
59 |
-
site_name, num_id = id_.
|
60 |
num_id = int(num_id)
|
61 |
_sites[site_name].append(num_id)
|
62 |
|
|
|
56 |
def _get_from_raw_ids(ids: List[str]) -> Dict[str, Image.Image]:
|
57 |
_sites = defaultdict(list)
|
58 |
for id_ in ids:
|
59 |
+
site_name, num_id = id_.rsplit('_', maxsplit=1)
|
60 |
num_id = int(num_id)
|
61 |
_sites[site_name].append(num_id)
|
62 |
|