Update app.py

app.py

@@ -266,7 +266,7 @@ def upload_to_ftp(filename):
         print(f"FTP upload error: {e}")
         
 def uploadNote(prompt,num_inference_steps,guidance_scale,timestamp):
-    filename= f'rv_C_{timestamp}.txt'
+    filename= f'rv_D_{timestamp}.txt'
     with open(filename, "w") as f:
         f.write(f"Realvis 5.0 (Tester D) \n")
         f.write(f"Date/time: {timestamp} \n")
@@ -370,7 +370,14 @@ def generate_60(
     rv_image = pipe(**batch_options).images[0]
     sd_image_path = f"rv_D_{timestamp}.png"
     rv_image.save(sd_image_path,optimize=False,compress_level=0)
-    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"rv_D_upscale_{timestamp}.png"
+    downscale1.save(downscale_path,optimize=False,compress_level=0)
+    pyx.upload_to_ftp(downscale_path) 
     unique_name = str(uuid.uuid4()) + ".png"  
     os.symlink(sd_image_path, unique_name)  
     return [unique_name]
@@ -411,7 +418,14 @@ def generate_90(
     rv_image = pipe(**batch_options).images[0]
     sd_image_path = f"rv_D_{timestamp}.png"
     rv_image.save(sd_image_path,optimize=False,compress_level=0)
-    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"rv_D_upscale_{timestamp}.png"
+    downscale1.save(downscale_path,optimize=False,compress_level=0)
+    pyx.upload_to_ftp(downscale_path) 
     unique_name = str(uuid.uuid4()) + ".png"  
     os.symlink(sd_image_path, unique_name)  
     return [unique_name]