Update app.py

app.py

@@ -153,10 +153,9 @@ def inference( prompt, negative_prompt, guidance_scale, ddim_steps, seed):
 
 @torch.no_grad()
 @spaces.GPU
-def edit_inference(prompt, negative_prompt, guidance_scale, ddim_steps, seed, start_noise, a1, a2, a3, a4):
+def edit_inference(network, prompt, negative_prompt, guidance_scale, ddim_steps, seed, start_noise, a1, a2, a3, a4):
     global device
     #global generator 
-    global network
     global unet
     global vae 
     global text_encoder
@@ -250,49 +249,7 @@ def sample_then_run():
     torch.save(network.proj, "model.pt" )
     return image, "model.pt"
 
-#@spaces.GPU
-def start_items():
-    print("Starting 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)
+
 
 class CustomImageDataset(Dataset):
     def __init__(self, images, transform=None):
@@ -528,7 +485,7 @@ with gr.Blocks(css="style.css") as demo:
         sample.click(fn=sample_then_run, outputs=[input_image, file_output])
         
         submit.click(
-            fn=edit_inference, inputs=[prompt, negative_prompt, cfg, steps, seed, injection_step, a1, a2, a3, a4], outputs=[gallery]
+            fn=edit_inference, inputs=[network, prompt, negative_prompt, cfg, steps, seed, injection_step, a1, a2, a3, a4], outputs=[gallery]
         )
         file_input.change(fn=file_upload, inputs=file_input, outputs = gallery)