Jyothirmai commited on
Commit
6ae0110
·
verified ·
1 Parent(s): 1618050

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def generate_caption_vitgpt(image):
20
  return caption
21
 
22
  def generate_caption_vitCoAtt(image):
23
- caption = tester.CaptionSampler.main(image)
24
  return caption
25
 
26
 
 
20
  return caption
21
 
22
  def generate_caption_vitCoAtt(image):
23
+ caption = ViTCoAtt.CaptionSampler.main(image)
24
  return caption
25
 
26