csgaobb commited on
Commit
26e5e7f
·
1 Parent(s): f2fb485

fix model loading bug

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -24,6 +24,7 @@ from argparse import ArgumentParser
24
  from torchvision.transforms.functional import pil_to_tensor
25
 
26
  from metauas import MetaUAS, set_random_seed, normalize, apply_ad_scoremap, safely_load_state_dict
 
27
 
28
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
29
 
 
24
  from torchvision.transforms.functional import pil_to_tensor
25
 
26
  from metauas import MetaUAS, set_random_seed, normalize, apply_ad_scoremap, safely_load_state_dict
27
+ from huggingface_hub import hf_hub_download
28
 
29
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
30