Update app.py

app.py

@@ -266,6 +266,13 @@ def infer(
     refine_path = f"sd35_refine_{seed}.png"
     refine.save(refine_path,optimize=False,compress_level=0)
     upload_to_ftp(refine_path)
+    with torch.no_grad():
+        upscale2 = upscaler_2(downscale, tiling=True, tile_width=1024, tile_height=1024)
+    print('-- got upscaled image --')
+    downscale2 = upscale2.resize((upscale2.width // 4, upscale2.height // 4),Image.LANCZOS)
+    upscale_path = f"sd35_upscale_{seed}.png"
+    downscale2.save(upscale_path,optimize=False,compress_level=0)
+    upload_to_ftp(upscale_path)
     return refine, seed, enhanced_prompt
 
 examples = [