Jyothirmai commited on
Commit
2bfbbec
·
verified ·
1 Parent(s): a4fcc26

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,11 +12,11 @@ from build_vocab import Vocabulary
12
 
13
  # Caption generation functions
14
  def generate_caption_clipgpt(image, max_tokens, temperature):
15
- caption = clipGPT.generate_caption_clipgpt(image)
16
  return caption
17
 
18
  def generate_caption_vitgpt(image, max_tokens, temperature):
19
- caption = vitGPT.generate_caption(image)
20
  return caption
21
 
22
  def generate_caption_vitCoAtt(image):
 
12
 
13
  # Caption generation functions
14
  def generate_caption_clipgpt(image, max_tokens, temperature):
15
+ caption = clipGPT.generate_caption_clipgpt(image, max_tokens, temperature)
16
  return caption
17
 
18
  def generate_caption_vitgpt(image, max_tokens, temperature):
19
+ caption = vitGPT.generate_caption(image, max_tokens, temperature)
20
  return caption
21
 
22
  def generate_caption_vitCoAtt(image):