Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -15,6 +15,13 @@ def read_image(image):
|
|
15 |
|
16 |
@spaces.GPU(duration=180)
|
17 |
def process_images(chapter_pages, character_bank_images, character_bank_names):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
18 |
chapter_pages = [read_image(image) for image in chapter_pages]
|
19 |
character_bank = {
|
20 |
"images": [read_image(image) for image in character_bank_images],
|
|
|
15 |
|
16 |
@spaces.GPU(duration=180)
|
17 |
def process_images(chapter_pages, character_bank_images, character_bank_names):
|
18 |
+
if chapter_pages is None:
|
19 |
+
return [], ""
|
20 |
+
if character_bank_images is None:
|
21 |
+
character_bank_images = []
|
22 |
+
character_bank_names = ""
|
23 |
+
if character_bank_names is None or character_bank_names == "":
|
24 |
+
character_bank_names = ",".join([os.path.splitext(os.path.basename(x))[0] for x in character_bank_images])
|
25 |
chapter_pages = [read_image(image) for image in chapter_pages]
|
26 |
character_bank = {
|
27 |
"images": [read_image(image) for image in character_bank_images],
|