Honzus24 commited on
Commit
1a1ed59
·
verified ·
1 Parent(s): 6274465

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -140,7 +140,8 @@ def flex_seq(input_seq, input_file):
140
  weights_path = get_weights_path(repo_id, file_weights)
141
 
142
  # Load weights
143
- state_dict = torch.load(weights_path, map_location=config['inference_args']['device']) model.load_state_dict(state_dict, strict=False)
 
144
  model.eval()
145
 
146
  data_to_collate = []
 
140
  weights_path = get_weights_path(repo_id, file_weights)
141
 
142
  # Load weights
143
+ state_dict = torch.load(weights_path, map_location=config['inference_args']['device'])
144
+ model.load_state_dict(state_dict, strict=False)
145
  model.eval()
146
 
147
  data_to_collate = []