dashakoryakovskaya commited on
Commit
4959d33
·
verified ·
1 Parent(s): 066685c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -143,7 +143,7 @@ def create_demo():
143
  if __name__ == "__main__":
144
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
145
  model = Mamba(num_layers = 2, d_input = 1024, d_model = 512, num_classes=7, model_name='jina', pooling=None).to(device)
146
- checkpoint = torch.load("models/Mamba_jina_checkpoint.pth", map_location=torch.device('cpu'))
147
  model.load_state_dict(checkpoint['model_state_dict'])
148
 
149
  demo = create_demo()
 
143
  if __name__ == "__main__":
144
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
145
  model = Mamba(num_layers = 2, d_input = 1024, d_model = 512, num_classes=7, model_name='jina', pooling=None).to(device)
146
+ checkpoint = torch.load("Mamba_jina_checkpoint.pth", map_location=torch.device('cpu'))
147
  model.load_state_dict(checkpoint['model_state_dict'])
148
 
149
  demo = create_demo()