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
|