usernameisanna commited on
Commit
93eb2df
·
verified ·
1 Parent(s): 41ac656

update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,7 +11,8 @@ processor = Blip2Processor.from_pretrained("Salesforce/blip2-flan-t5-xl")
11
  # model_path = "full-blip2-deit-config-yes-no-2.pth"
12
  # model = torch.load("./full-blip2-deit-config-2.pth")
13
  # model = torch.load("./full-blip2-deit.pth") # not working - error
14
- model = torch.load("./full-blip2-deit-config-free-form-4-ver-2.pth")
 
15
 
16
  model.eval() # Set the model to evaluation mode
17
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
11
  # model_path = "full-blip2-deit-config-yes-no-2.pth"
12
  # model = torch.load("./full-blip2-deit-config-2.pth")
13
  # model = torch.load("./full-blip2-deit.pth") # not working - error
14
+ # model = torch.load("./full-blip2-deit-config-free-form-4-ver-2.pth")
15
+ model = torch.load("./full_config_blip2-deit-05")
16
 
17
  model.eval() # Set the model to evaluation mode
18
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")