kasun commited on
Commit
1e1dd5f
1 Parent(s): 7ab6979

fixed a bug

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -43,12 +43,12 @@ vitgpt_tokenizer = AutoTokenizer.from_pretrained("nlpconnect/vit-gpt2-image-capt
43
 
44
  device = "cuda" if torch.cuda.is_available() else "cpu"
45
 
46
- # git_model_base.to(device)
47
  blip_model_base.to(device)
48
- # git_model_large_coco.to(device)
49
- # git_model_large_textcaps.to(device)
50
- # blip_model_large.to(device)
51
- # vitgpt_model.to(device)
52
  # coca_model.to(device)
53
  # blip2_model.to(device)
54
 
 
43
 
44
  device = "cuda" if torch.cuda.is_available() else "cpu"
45
 
46
+ git_model_base.to(device)
47
  blip_model_base.to(device)
48
+ git_model_large_coco.to(device)
49
+ git_model_large_textcaps.to(device)
50
+ blip_model_large.to(device)
51
+ vitgpt_model.to(device)
52
  # coca_model.to(device)
53
  # blip2_model.to(device)
54