1inkusFace commited on
Commit
bc256b2
·
verified ·
1 Parent(s): 423ee98

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -263,7 +263,7 @@ def captioning(img):
263
  "Describe this image."
264
  )
265
  inputsa = processor5(images=img, text=cap_prompt, return_tensors="pt").to('cuda')
266
- inputsb = processor5(images=img, text=cap_prompt, return_tensors="pt").to('cuda')
267
  '''
268
  generated_ids = model5.generate(
269
  **inputsa,
 
263
  "Describe this image."
264
  )
265
  inputsa = processor5(images=img, text=cap_prompt, return_tensors="pt").to('cuda')
266
+ inputsb = processor5(images=img, text=cap_prompt2, return_tensors="pt").to('cuda')
267
  '''
268
  generated_ids = model5.generate(
269
  **inputsa,