Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -19,10 +19,9 @@ def generate(model_name, image, text):
|
|
19 |
# print("Model_name:", model_name)
|
20 |
print("Image:", image)
|
21 |
print("Text:", text)
|
22 |
-
# model_pipe.to("cuda")
|
23 |
model_pipe = VQAScore(model="clip-flant5-xl") # our recommended scoring model
|
24 |
-
print("Model initialized, now moving to cuda")
|
25 |
-
model_pipe.to("cuda")
|
26 |
print("Generating!")
|
27 |
return model_pipe(images=[image], texts=[text])
|
28 |
|
|
|
19 |
# print("Model_name:", model_name)
|
20 |
print("Image:", image)
|
21 |
print("Text:", text)
|
|
|
22 |
model_pipe = VQAScore(model="clip-flant5-xl") # our recommended scoring model
|
23 |
+
# print("Model initialized, now moving to cuda")
|
24 |
+
# model_pipe.to("cuda")
|
25 |
print("Generating!")
|
26 |
return model_pipe(images=[image], texts=[text])
|
27 |
|