Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -9,9 +9,9 @@ from t2v_metrics import VQAScore, list_all_vqascore_models
|
|
9 |
print(list_all_vqascore_models())
|
10 |
|
11 |
# Initialize the model only once
|
12 |
-
if torch.cuda.is_available():
|
13 |
-
|
14 |
-
|
15 |
|
16 |
@spaces.GPU
|
17 |
def generate(model_name, image, text):
|
|
|
9 |
print(list_all_vqascore_models())
|
10 |
|
11 |
# Initialize the model only once
|
12 |
+
# if torch.cuda.is_available():
|
13 |
+
model_pipe = VQAScore(model="clip-flant5-xl") # our recommended scoring model
|
14 |
+
print("Model initialized!")
|
15 |
|
16 |
@spaces.GPU
|
17 |
def generate(model_name, image, text):
|