ragavsachdeva commited on
Commit
80a0d81
·
verified ·
1 Parent(s): 5c52575

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
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],