zhiqiulin commited on
Commit
f5adbef
·
verified ·
1 Parent(s): 9910355

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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
  print("Generating!")
25
  return model_pipe(images=[image], texts=[text])
26
 
 
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])
27