Spaces:

maze
/

FastStyleTransfer

Runtime error

maze commited on Feb 17, 2022

Commit

7332f3d

1 Parent(s): 1238ccd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -136,7 +136,7 @@ def process(image, model):
 def main(image, backbone, style):
-    transformer.load_state_dict(torch.load(Rain_Princess))
     image = Image.fromarray(image)
     isize = image.size
     image = process(image, transformer)

 def main(image, backbone, style):
+    transformer.load_state_dict(torch.load(Rain_Princess, map_location=torch.device('cpu')))
     image = Image.fromarray(image)
     isize = image.size
     image = process(image, transformer)