1inkusFace commited on
Commit
fd60bf9
·
verified ·
1 Parent(s): c12d5e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -290,6 +290,7 @@ def captioning(img):
290
  output_prompt.append(generated_text)
291
  print(generated_text)
292
  # Loop through prompts array:
 
293
  for prompt in prompts_array:
294
  inputs = processor5(images=img, text=prompt, return_tensors="pt").to('cuda')
295
  #with torch.no_grad():
@@ -328,6 +329,7 @@ def captioning(img):
328
  response_text = generated_texta.replace(generated_text, "").strip()
329
  print(f"{response_text}\n")
330
  output_prompt.append(response_text)
 
331
  output_prompt = " ".join(output_prompt)
332
  return output_prompt
333
 
 
290
  output_prompt.append(generated_text)
291
  print(generated_text)
292
  # Loop through prompts array:
293
+ '''
294
  for prompt in prompts_array:
295
  inputs = processor5(images=img, text=prompt, return_tensors="pt").to('cuda')
296
  #with torch.no_grad():
 
329
  response_text = generated_texta.replace(generated_text, "").strip()
330
  print(f"{response_text}\n")
331
  output_prompt.append(response_text)
332
+ '''
333
  output_prompt = " ".join(output_prompt)
334
  return output_prompt
335