Update app.py

app.py

@@ -107,9 +107,9 @@ import gc
 import torch.nn.functional as F
 from sageattention import sageattn
 
-
+'''
 import pythran
-
+paramiko_include_dir = paramiko.get_include()
 with open("pyt.py", "w") as f:
     f.write("""
 import paramiko
@@ -135,12 +135,11 @@ def upload_to_ftp(filename):
         print(f"FTP upload error: {e}")
         
     """)
-    
+'''
 os.environ['DISTUTILS_USE_SDK'] = '1'  # Force setuptools (distutils) to use the installed SDK.
-subprocess.run(['pythran', './pyt.py', '-DUSE_XSIMD', '-fopenmp', '-march=native'])
+subprocess.run(['pythran', './pyt.py', '-DUSE_XSIMD', '-fopenmp', '-march=native','-I' + paramiko_include_dir])
 import pyt
 
-
 torch.backends.cuda.matmul.allow_tf32 = False
 torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
 torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
@@ -468,14 +467,14 @@ def generate_30(
     rv_image = pipe(**batch_options).images[0]
     sd_image_path = f"rv_B_{timestamp}.png"
     rv_image.save(sd_image_path,optimize=False,compress_level=0)
-    pyt.upload_to_ftp(sd_image_path)    
+    pyx.upload_to_ftp(sd_image_path)    
     torch.set_float32_matmul_precision("medium")
     with torch.no_grad():
         upscale = upscaler(rv_image, tiling=True, tile_width=256, tile_height=256)
     downscale1 = upscale.resize((upscale.width // 4, upscale.height // 4), Image.LANCZOS)
     downscale_path = f"rv50_upscale_{timestamp}.png"
     downscale1.save(downscale_path,optimize=False,compress_level=0)
-    pyt.upload_to_ftp(downscale_path) 
+    pyx.upload_to_ftp(downscale_path) 
     unique_name = str(uuid.uuid4()) + ".png"  
     os.symlink(sd_image_path, unique_name)  
     return [unique_name]