zhiqiulin commited on
Commit
42e3288
·
verified ·
1 Parent(s): 2f099c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -19,10 +19,9 @@ def generate(model_name, image, text):
19
  # print("Model_name:", model_name)
20
  print("Image:", image)
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])
28
 
 
19
  # print("Model_name:", model_name)
20
  print("Image:", image)
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