Spaces:
Sleeping
Sleeping
usernameisanna
commited on
Commit
•
e6174de
1
Parent(s):
4fe2fbf
update path for free form
Browse files
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")
|