zhiqiulin commited on
Commit
a4aa340
·
verified ·
1 Parent(s): 6922f48

Update app.py

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