ajayarora1235 commited on
Commit
971afa1
·
1 Parent(s): f9f05d9

correctly named hubert pth

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -246,7 +246,7 @@ associated_links = {}
246
  def load_hubert():
247
  global hubert_model
248
  # Load the model
249
- hubert_model = torch.load("hubert_base.pth", map_location=config.device)
250
 
251
  # Prepare the model
252
  hubert_model = hubert_model.to(config.device)
 
246
  def load_hubert():
247
  global hubert_model
248
  # Load the model
249
+ hubert_model = torch.load("hubert.pth", map_location=config.device)
250
 
251
  # Prepare the model
252
  hubert_model = hubert_model.to(config.device)