zhiqiulin commited on
Commit
4536949
·
verified ·
1 Parent(s): e58c712

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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