stzhao commited on
Commit
64bbe5e
·
verified ·
1 Parent(s): a10923f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def load_models():
19
 
20
  model, tokenizer = load_models()
21
 
22
- @spaces.GPU()
23
  def generate_enhanced_caption(image_caption, text_caption):
24
  """Generate enhanced caption using the LeX-Enhancer model"""
25
  combined_caption = f"{image_caption}, with the text on it: {text_caption}."
 
19
 
20
  model, tokenizer = load_models()
21
 
22
+ # @spaces.GPU()
23
  def generate_enhanced_caption(image_caption, text_caption):
24
  """Generate enhanced caption using the LeX-Enhancer model"""
25
  combined_caption = f"{image_caption}, with the text on it: {text_caption}."