Hugging Face's logo

Spaces:
CVPR
/
BigDL-Nano_inference
Runtime error

App Files Community
1
rnwang commited on
Commit
621fa8e
·
1 Parent(s): f9cae44

fix

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -55,9 +55,9 @@ import huggingface_hub
55
 
56
  device = 'cpu'
57
  dtype = torch.float32
58
- MODEL_REPO = 'CVPR/generator'
59
  ckpt_path = huggingface_hub.hf_hub_download(
60
- MODEL_REPO, f'generator.pt')
61
  generator = torch.load(ckpt_path)
62
  generator.eval()
63
  generator.to(device, dtype)
 
55
 
56
  device = 'cpu'
57
  dtype = torch.float32
58
+ MODEL_REPO = 'CVPR/FSPBT'
59
  ckpt_path = huggingface_hub.hf_hub_download(
60
+ MODEL_REPO, 'generator.pt')
61
  generator = torch.load(ckpt_path)
62
  generator.eval()
63
  generator.to(device, dtype)