padmanabhbosamia commited on
Commit
b792a8c
·
verified ·
1 Parent(s): ead853e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ def load_trained_model():
24
  filename="best_model_compressed.pt",
25
  token=os.getenv('HF_TOKEN')
26
  )
27
- checkpoint = torch.load(model_path, map_location=config.device, weights_only=True)
28
  model.load_state_dict(checkpoint['model_state_dict'])
29
  model.to(config.device)
30
  model.eval()
 
24
  filename="best_model_compressed.pt",
25
  token=os.getenv('HF_TOKEN')
26
  )
27
+ checkpoint = torch.load(model_path, map_location=config.device)
28
  model.load_state_dict(checkpoint['model_state_dict'])
29
  model.to(config.device)
30
  model.eval()