Hugging Face's logo

Spaces:
befozg
/
stylematte
Runtime error

App Files Community
1
befozg commited on
Commit
39fedb9
·
1 Parent(s): 8f2bc26

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from test import inference_img
4
  device='cuda'
5
  model = MaskForm()
6
  model = model.to(device)
7
- checkpoint = f"stylematte_synth.pth"
8
  state_dict = torch.load(checkpoint, map_location=f'{device}')
9
 
10
  model.load_state_dict(state_dict)
 
4
  device='cuda'
5
  model = MaskForm()
6
  model = model.to(device)
7
+ checkpoint = f"stylematte.pth"
8
  state_dict = torch.load(checkpoint, map_location=f'{device}')
9
 
10
  model.load_state_dict(state_dict)