Spaces:
Runtime error
Runtime error
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],
|