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
|