usernameisanna commited on
Commit
e6174de
1 Parent(s): 4fe2fbf

update path for free form

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