Update app.py

app.py

@@ -20,15 +20,19 @@ from huggingface_hub import snapshot_download
 import spaces
 
 
-gr.State(generator)
-gr.State(unet)
-gr.State(vae)
-gr.State(text_encoder)
-gr.State(tokenizer)
-gr.State(noise_scheduler)
-gr.State(network)
+generator = gr.State()
+unet = gr.State()
+vae = gr.State()
+text_encoder = gr.State()
+tokenizer = gr.State()
+noise_scheduler = gr.State()
+network = gr.State()
 device = gr.State("cuda")
 #generator = torch.Generator(device=device)
+young = gr.State()
+pointy = gr.State()
+wavy = gr.State()
+thick = gr.State()
 
 models_path = snapshot_download(repo_id="Snapchat/w2w")
 
@@ -43,10 +47,7 @@ pinverse = torch.load(f"{models_path}/files/pinverse_1000pc.pt", map_location=to
 unet, vae, text_encoder, tokenizer, noise_scheduler = load_models(device)
 
 
-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)