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