Spaces:

seddiktrk
/

CLIP-GPT_Image_Captionning

Sleeping

seddiktrk commited on Aug 31, 2024

Commit

0aa5621

verified ·

1 Parent(s): c4633c3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -169,7 +169,7 @@ print('loading model')
 print()
 ## Prepare Model
 CliPGPT = ClipCaptionModel()
-path = "model_epoch_1.pt"
 state_dict = torch.load(path,map_location=torch.device('cpu'))
 # Apply the weights to the model

 print()
 ## Prepare Model
 CliPGPT = ClipCaptionModel()
+path = "model_epoch_1_loss_2.0695.pt"
 state_dict = torch.load(path,map_location=torch.device('cpu'))
 # Apply the weights to the model