AP123 commited on
Commit
a68f6ce
·
verified ·
1 Parent(s): d301996

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -48,7 +48,7 @@ def transform_image(face_image):
48
  processed_face_tensor = transforms.ToTensor()(processed_face_image).unsqueeze(0).to("cuda")
49
 
50
  # Load the style image from the local path, resize it and convert to tensor
51
- style_image_path = "examples/soyjak2.jpeg" # Ensure this path is correct
52
  style_image = Image.open(style_image_path).resize(desired_size, Image.LANCZOS)
53
  style_image_tensor = transforms.ToTensor()(style_image).unsqueeze(0).to("cuda")
54
 
 
48
  processed_face_tensor = transforms.ToTensor()(processed_face_image).unsqueeze(0).to("cuda")
49
 
50
  # Load the style image from the local path, resize it and convert to tensor
51
+ style_image_path = "examples/soyjak2.jpg" # Ensure this path is correct
52
  style_image = Image.open(style_image_path).resize(desired_size, Image.LANCZOS)
53
  style_image_tensor = transforms.ToTensor()(style_image).unsqueeze(0).to("cuda")
54