1inkusFace commited on
Commit
8cd1460
·
verified ·
1 Parent(s): 6e68cd9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -311,15 +311,15 @@ def captioning(img, prompt_1,prompt_2,prompt_3,prompt_4,prompt_5):
311
  output_prompt.append(generated_text)
312
  print(generated_text)
313
  prompt_array=[]
314
- if prompt_1 in not None:
315
  prompt_array.append(prompt_1)
316
- if prompt_2 in not None:
317
  prompt_array.append(prompt_2)
318
- if prompt_3 in not None:
319
  prompt_array.append(prompt_3)
320
- if prompt_4 in not None:
321
  prompt_array.append(prompt_4)
322
- if prompt_5 in not None:
323
  prompt_array.append(prompt_5)
324
  for prompt in prompts_array:
325
  inputs = processor5(images=img, text=prompt, return_tensors="pt").to('cuda')
 
311
  output_prompt.append(generated_text)
312
  print(generated_text)
313
  prompt_array=[]
314
+ if prompt_1 is not None:
315
  prompt_array.append(prompt_1)
316
+ if prompt_2 is not None:
317
  prompt_array.append(prompt_2)
318
+ if prompt_3 is not None:
319
  prompt_array.append(prompt_3)
320
+ if prompt_4 is not None:
321
  prompt_array.append(prompt_4)
322
+ if prompt_5 is not None:
323
  prompt_array.append(prompt_5)
324
  for prompt in prompts_array:
325
  inputs = processor5(images=img, text=prompt, return_tensors="pt").to('cuda')