Pix2Pix-Video

Running on Zero

fffiloni commited on Feb 9, 2023

Commit

82002ed

•

1 Parent(s): c3d5a7d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,6 +44,7 @@ def pix2pix(
     generator = torch.Generator("cuda").manual_seed(seed)
     try:
         ratio = min(height / image.height, width / image.width)
         image = image.resize((int(image.width * ratio), int(image.height * ratio)), Image.LANCZOS)

     generator = torch.Generator("cuda").manual_seed(seed)
     try:
+        image = Image.open(image)
         ratio = min(height / image.height, width / image.width)
         image = image.resize((int(image.width * ratio), int(image.height * ratio)), Image.LANCZOS)