Update app.py
Browse files
app.py
CHANGED
|
@@ -169,7 +169,7 @@ class ClipCaptionModel(nn.Module):
|
|
| 169 |
|
| 170 |
## Prepare Model
|
| 171 |
CliPGPT = ClipCaptionModel()
|
| 172 |
-
path = "
|
| 173 |
state_dict = torch.load(path)
|
| 174 |
|
| 175 |
# Apply the weights to the model
|
|
|
|
| 169 |
|
| 170 |
## Prepare Model
|
| 171 |
CliPGPT = ClipCaptionModel()
|
| 172 |
+
path = "model_epoch_1.pt"
|
| 173 |
state_dict = torch.load(path)
|
| 174 |
|
| 175 |
# Apply the weights to the model
|