Update app.py

app.py

@@ -7,7 +7,7 @@ import numpy as np
 import torch
 from torch import nn
 print(torch.__version__)
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+device = torch.device('cpu')
 print(device)
 
 print('importing tokenizer')
@@ -166,13 +166,14 @@ class ClipCaptionModel(nn.Module):
 
 
 print('loading model')
+print()
 ## Prepare Model
 CliPGPT = ClipCaptionModel()
 path = "model_epoch_1.pt"
-state_dict = torch.load(path)
+state_dict = torch.load(path,map_location=torch.device('cpu')))
 
 # Apply the weights to the model
-CliPGPT.load_state_dict(state_dict,map_location=torch.device('cpu') )
+CliPGPT.load_state_dict(state_dict)
 CliPGPT.to(device)
 
 print('importing CLIP')