Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -21,6 +21,7 @@ def generate(model_name, image, text):
|
|
21 |
print("Text:", text)
|
22 |
# model_pipe.to("cuda")
|
23 |
model_pipe = VQAScore(model="clip-flant5-xl") # our recommended scoring model
|
|
|
24 |
model_pipe.to("cuda")
|
25 |
print("Generating!")
|
26 |
return model_pipe(images=[image], texts=[text])
|
|
|
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])
|