zhiqiulin commited on
Commit
04cc94e
1 Parent(s): 8e64bf0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from t2v_metrics import VQAScore, list_all_vqascore_models
8
  # Global model variable, but do not initialize or move to CUDA here
9
  model_pipe = VQAScore(model="clip-flant5-xl", device="cuda") # our recommended scoring model
10
 
11
- @spaces.GPU(duration = 300)
12
  def generate(model_name, image, text):
13
  print(list_all_vqascore_models()) # Debug: List available models
14
  print("Image:", image) # Debug: Print image path
 
8
  # Global model variable, but do not initialize or move to CUDA here
9
  model_pipe = VQAScore(model="clip-flant5-xl", device="cuda") # our recommended scoring model
10
 
11
+ @spaces.GPU(duration = 20)
12
  def generate(model_name, image, text):
13
  print(list_all_vqascore_models()) # Debug: List available models
14
  print("Image:", image) # Debug: Print image path