Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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="
|
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):
|