Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -321,7 +321,7 @@ def captioning(img, prompt_1,prompt_2,prompt_3,prompt_4,prompt_5):
|
|
321 |
prompt_array.append(prompt_4)
|
322 |
if prompt_5 is not None:
|
323 |
prompt_array.append(prompt_5)
|
324 |
-
for prompt in
|
325 |
inputs = processor5(images=img, text=prompt, return_tensors="pt").to('cuda')
|
326 |
#with torch.no_grad():
|
327 |
generated_ids = model5.generate(
|
|
|
321 |
prompt_array.append(prompt_4)
|
322 |
if prompt_5 is not None:
|
323 |
prompt_array.append(prompt_5)
|
324 |
+
for prompt in prompt_array:
|
325 |
inputs = processor5(images=img, text=prompt, return_tensors="pt").to('cuda')
|
326 |
#with torch.no_grad():
|
327 |
generated_ids = model5.generate(
|