yuntian-deng commited on
Commit
cfbf9d9
·
1 Parent(s): 1729d9d

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -6,7 +6,7 @@ import numpy as np
6
  from PIL import Image
7
  from huggingface_hub import hf_hub_download
8
  import json
9
- import OS
10
 
11
  def load_model_from_config(config_path, model_name, device='cuda'):
12
  # Load the config file
 
6
  from PIL import Image
7
  from huggingface_hub import hf_hub_download
8
  import json
9
+ import os
10
 
11
  def load_model_from_config(config_path, model_name, device='cuda'):
12
  # Load the config file