Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ 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
|
|
|
10 |
|
11 |
# Initialize the model only once
|
12 |
# if torch.cuda.is_available():
|
13 |
+
model_pipe = VQAScore(model="clip-flant5-xl", device="cpu") # our recommended scoring model
|
14 |
print("Model initialized!")
|
15 |
|
16 |
@spaces.GPU
|