Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,7 @@ def generate(model_name, image, text):
|
|
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 |
|
|
|
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 |
|