zhiqiulin commited on
Commit
98d29ed
1 Parent(s): 5041f6c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ torch.jit.script = lambda f: f # Avoid script error in lambda
7
  from t2v_metrics import VQAScore, list_all_vqascore_models
8
 
9
  # Global model variable, but do not initialize or move to CUDA here
10
- model_pipe = VQAScore(model="clip-flant5-xl", device="cpu") # our recommended scoring model
11
 
12
  @spaces.GPU
13
  def generate(model_name, image, text):
 
7
  from t2v_metrics import VQAScore, list_all_vqascore_models
8
 
9
  # Global model variable, but do not initialize or move to CUDA here
10
+ model_pipe = VQAScore(model="clip-flant5-xl", device="cuda") # our recommended scoring model
11
 
12
  @spaces.GPU
13
  def generate(model_name, image, text):