Update app.py

app.py

@@ -40,8 +40,7 @@ v = torch.load(f"{models_path}/files/V.pt", map_location=torch.device('cpu')).bf
 proj = torch.load(f"{models_path}/files/proj_1000pc.pt", map_location=torch.device('cpu')).bfloat16().to(device)
 df = torch.load(f"{models_path}/files/identity_df.pt")
 weight_dimensions = torch.load(f"{models_path}/files/weight_dimensions.pt")
-pinverse = torch.load(f"{models_path}/files/pinverse_1000pc.pt", map_location=torch.device('cpu'))#.bfloat16()#.to(device)
-
+pinverse = torch.load(f"{models_path}/files/pinverse_1000pc.pt", map_location=torch.device('cpu')).bfloat16().to(device)
 
 unet, vae, text_encoder, tokenizer, noise_scheduler = load_models(device)
 
@@ -206,52 +205,52 @@ def sample_then_run():
     torch.save(network.proj, "model.pt" )
     return image, "model.pt"
 
-
-global young
-global pointy
-global wavy 
-global thick
-
-young = get_direction(df, "Young", pinverse, 1000, device)
-young = debias(young, "Male", df, pinverse, device)
-young = debias(young, "Pointy_Nose", df, pinverse, device)
-young = debias(young, "Wavy_Hair", df, pinverse, device)
-young = debias(young, "Chubby", df, pinverse, device)
-young = debias(young, "No_Beard", df, pinverse, device)
-young = debias(young, "Mustache", df, pinverse, device)
-
-
-pointy = get_direction(df, "Pointy_Nose", pinverse, 1000, device)
-pointy = debias(pointy, "Young", df, pinverse, device)
-pointy = debias(pointy, "Male", df, pinverse, device)
-pointy = debias(pointy, "Wavy_Hair", df, pinverse, device)
-pointy = debias(pointy, "Chubby", df, pinverse, device)
-pointy = debias(pointy, "Heavy_Makeup", df, pinverse, device)
-
-
-
-wavy = get_direction(df, "Wavy_Hair", pinverse, 1000, device)
-wavy = debias(wavy, "Young", df, pinverse, device)
-wavy = debias(wavy, "Male", df, pinverse, device)
-wavy = debias(wavy, "Pointy_Nose", df, pinverse, device)
-wavy = debias(wavy, "Chubby", df, pinverse, device)
-wavy = debias(wavy, "Heavy_Makeup", df, pinverse, device)
-
-
-thick = get_direction(df, "Bushy_Eyebrows", pinverse, 1000, device)
-thick = debias(thick, "Male", df, pinverse, device)
-thick = debias(thick, "Young", df, pinverse, device)
-thick = debias(thick, "Pointy_Nose", df, pinverse, device)
-thick = debias(thick, "Wavy_Hair", df, pinverse, device)
-thick = debias(thick, "Mustache", df, pinverse, device)
-thick = debias(thick, "No_Beard", df, pinverse, device)
-thick = debias(thick, "Sideburns", df, pinverse, device)
-thick = debias(thick, "Big_Nose", df, pinverse, device)
-thick = debias(thick, "Big_Lips", df, pinverse, device)
-thick = debias(thick, "Black_Hair", df, pinverse, device)
-thick = debias(thick, "Brown_Hair", df, pinverse, device)
-thick = debias(thick, "Pale_Skin", df, pinverse, device)
-thick = debias(thick, "Heavy_Makeup", df, pinverse, device)
+@spaces.GPU
+def start_items():
+    global young
+    global pointy
+    global wavy 
+    global thick
+    young = get_direction(df, "Young", pinverse, 1000, device)
+    young = debias(young, "Male", df, pinverse, device)
+    young = debias(young, "Pointy_Nose", df, pinverse, device)
+    young = debias(young, "Wavy_Hair", df, pinverse, device)
+    young = debias(young, "Chubby", df, pinverse, device)
+    young = debias(young, "No_Beard", df, pinverse, device)
+    young = debias(young, "Mustache", df, pinverse, device)
+    
+    
+    pointy = get_direction(df, "Pointy_Nose", pinverse, 1000, device)
+    pointy = debias(pointy, "Young", df, pinverse, device)
+    pointy = debias(pointy, "Male", df, pinverse, device)
+    pointy = debias(pointy, "Wavy_Hair", df, pinverse, device)
+    pointy = debias(pointy, "Chubby", df, pinverse, device)
+    pointy = debias(pointy, "Heavy_Makeup", df, pinverse, device)
+    
+    
+    
+    wavy = get_direction(df, "Wavy_Hair", pinverse, 1000, device)
+    wavy = debias(wavy, "Young", df, pinverse, device)
+    wavy = debias(wavy, "Male", df, pinverse, device)
+    wavy = debias(wavy, "Pointy_Nose", df, pinverse, device)
+    wavy = debias(wavy, "Chubby", df, pinverse, device)
+    wavy = debias(wavy, "Heavy_Makeup", df, pinverse, device)
+    
+    
+    thick = get_direction(df, "Bushy_Eyebrows", pinverse, 1000, device)
+    thick = debias(thick, "Male", df, pinverse, device)
+    thick = debias(thick, "Young", df, pinverse, device)
+    thick = debias(thick, "Pointy_Nose", df, pinverse, device)
+    thick = debias(thick, "Wavy_Hair", df, pinverse, device)
+    thick = debias(thick, "Mustache", df, pinverse, device)
+    thick = debias(thick, "No_Beard", df, pinverse, device)
+    thick = debias(thick, "Sideburns", df, pinverse, device)
+    thick = debias(thick, "Big_Nose", df, pinverse, device)
+    thick = debias(thick, "Big_Lips", df, pinverse, device)
+    thick = debias(thick, "Black_Hair", df, pinverse, device)
+    thick = debias(thick, "Brown_Hair", df, pinverse, device)
+    thick = debias(thick, "Pale_Skin", df, pinverse, device)
+    thick = debias(thick, "Heavy_Makeup", df, pinverse, device)