Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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
|
315 |
prompt_array.append(prompt_1)
|
316 |
-
if prompt_2
|
317 |
prompt_array.append(prompt_2)
|
318 |
-
if prompt_3
|
319 |
prompt_array.append(prompt_3)
|
320 |
-
if prompt_4
|
321 |
prompt_array.append(prompt_4)
|
322 |
-
if prompt_5
|
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')
|