Hugging Face's logo

Spaces:
opetrova
/
face-frontalization
Runtime error

App Files Community
3

Update app.py

#2
by soiz1 - opened
base: refs/heads/main
←
from: refs/pr/2
Discussion Files changed
+1
-1
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ network_url = hf_hub_url(repo_id="opetrova/face-frontalization", filename="netwo
20
  r = requests.get(network_url, allow_redirects=True)
21
  open('network.py', 'wb').write(r.content)
22
 
23
- saved_model = torch.load(path_to_model, map_location=torch.device('cpu'))
24
 
25
  def frontalize(image):
26
 
 
20
  r = requests.get(network_url, allow_redirects=True)
21
  open('network.py', 'wb').write(r.content)
22
 
23
+ saved_model = torch.load(path_to_model, map_location=torch.device("cpu"), weights_only=False)
24
 
25
  def frontalize(image):
26