Update app.py
Browse files
app.py
CHANGED
@@ -170,7 +170,7 @@ print()
|
|
170 |
## Prepare Model
|
171 |
CliPGPT = ClipCaptionModel()
|
172 |
path = "model_epoch_1.pt"
|
173 |
-
state_dict = torch.load(path,map_location=torch.device('cpu'))
|
174 |
|
175 |
# Apply the weights to the model
|
176 |
CliPGPT.load_state_dict(state_dict)
|
|
|
170 |
## Prepare Model
|
171 |
CliPGPT = ClipCaptionModel()
|
172 |
path = "model_epoch_1.pt"
|
173 |
+
state_dict = torch.load(path,map_location=torch.device('cpu'))
|
174 |
|
175 |
# Apply the weights to the model
|
176 |
CliPGPT.load_state_dict(state_dict)
|