Update app.py

app.py

@@ -19,15 +19,15 @@ from lora_w2w import LoRAw2w
 from huggingface_hub import snapshot_download
 import spaces
 
-global device
-global generator 
-global unet
-global vae 
-global text_encoder
-global tokenizer
-global noise_scheduler
-global network
-device = "cuda"
+
+gr.State(generator)
+gr.State(unet)
+gr.State(vae)
+gr.State(text_encoder)
+gr.State(tokenizer)
+gr.State(noise_scheduler)
+gr.State(network)
+device = gr.State("cuda")
 #generator = torch.Generator(device=device)
 
 models_path = snapshot_download(repo_id="Snapchat/w2w")
@@ -43,10 +43,10 @@ pinverse = torch.load(f"{models_path}/files/pinverse_1000pc.pt", map_location=to
 unet, vae, text_encoder, tokenizer, noise_scheduler = load_models(device)
 
 
-global young
-global pointy
-global wavy 
-global thick
+gr.State(young)
+gr.State(pointy)
+gr.State(wavy) 
+gr.State(thick)
 
 young = get_direction(df, "Young", pinverse, 1000, device)
 young = debias(young, "Male", df, pinverse, device)