1inkusFace commited on
Commit
4be7938
·
verified ·
1 Parent(s): 8cd1460

Update app.py

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