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

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
  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])