Spaces:

Legola
/

StyleGAN

Build error

Legola commited on May 10, 2023

Commit

58cbf33

1 Parent(s): a6f7fc9

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -77,7 +77,7 @@ def image_transform(image):
         image = image.convert('RGB')
     # Apply the same transformations as before
-    image = transform(image).unsqueeze(0)
     return image

         image = image.convert('RGB')
     # Apply the same transformations as before
+    image =image_transform(image).unsqueeze(0)
     return image