Update app.py

app.py

@@ -77,6 +77,7 @@ styles = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in style_list}
 DEFAULT_STYLE_NAME = "Style Zero"
 STYLE_NAMES = list(styles.keys())
 HF_TOKEN = os.getenv("HF_TOKEN")
+os.putenv('TORCH_LINALG_PREFER_CUSOLVER','1')
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
@@ -198,7 +199,7 @@ def uploadNote(prompt,num_inference_steps,guidance_scale,timestamp):
         f.write(f"Model Scheduler: Euler_a all_custom before cuda \n")
         f.write(f"Model VAE: sdxl-vae-bf16\n")
         f.write(f"To cuda and bfloat \n")
-    upload_to_ftp(filename) 
+    return filename
 '''
 
 pyx = cyper.inline(code)
@@ -234,7 +235,8 @@ def generate_30(
         options["use_resolution_binning"] = True
     images = []
     timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
-    pyx.func(prompt,num_inference_steps,guidance_scale,timestamp)
+    filename = pyx.func(prompt,num_inference_steps,guidance_scale,timestamp)
+    upload_to_ftp(filename) 
     #uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
     batch_options = options.copy()
     rv_image = pipe(**batch_options).images[0]