Hugging Face's logo

Spaces:
snap-research
/
weights2weights
Running on Zero

App Files Community
5
amildravid4292 commited on
Commit
95ecabf
·
verified ·
1 Parent(s): ad7da82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -7
app.py CHANGED
@@ -42,30 +42,33 @@ pinverse = torch.load(f"{models_path}/files/pinverse_1000pc.pt", map_location=to
42
 
43
  unet.value, vae.value, text_encoder.value, tokenizer.value, noise_scheduler.value = load_models(device.value)
44
 
45
-
46
- gr.State(young) = get_direction(df, "Young", pinverse, 1000, device.value)
47
  young.value = debias(young.value, "Male", df, pinverse, device.value)
48
  young.value = debias(young.value, "Pointy_Nose", df, pinverse, device.value)
49
  young.value = debias(young.value, "Wavy_Hair", df, pinverse, device.value)
50
  young.value = debias(young.value, "Chubby", df, pinverse, device.value)
51
  young.value = debias(young.value, "No_Beard", df, pinverse, device.value)
52
  young.value = debias(young.value, "Mustache", df, pinverse, device.value)
53
-
54
- gr.State(pointy) = get_direction(df, "Pointy_Nose", pinverse, 1000, device.value)
 
55
  pointy.value = debias(pointy.value, "Young", df, pinverse, device.value)
56
  pointy.value = debias(pointy.value, "Male", df, pinverse, device.value)
57
  pointy.value = debias(pointy.value, "Wavy_Hair", df, pinverse, device.value)
58
  pointy.value = debias(pointy.value, "Chubby", df, pinverse, device.value)
59
  pointy.value = debias(pointy.value, "Heavy_Makeup", df, pinverse, device.value)
60
-
61
- gr.State(wavy) = get_direction(df, "Wavy_Hair", pinverse, 1000, device.value)
 
62
  wavy.value = debias(wavy.value, "Young", df, pinverse, device.value)
63
  wavy.value = debias(wavy.value, "Male", df, pinverse, device.value)
64
  wavy.value = debias(wavy.value, "Pointy_Nose", df, pinverse, device.value)
65
  wavy.value = debias(wavy.value, "Chubby", df, pinverse, device.value)
66
  wavy.value = debias(wavy.value, "Heavy_Makeup", df, pinverse, device.value)
67
 
68
- gr.State(thick) = get_direction(df, "Bushy_Eyebrows", pinverse, 1000, device.value)
 
69
  thick.value = debias(thick.value, "Male", df, pinverse, device.value)
70
  thick.value = debias(thick.value, "Young", df, pinverse, device.value)
71
  thick.value = debias(thick.value, "Pointy_Nose", df, pinverse, device.value)
 
42
 
43
  unet.value, vae.value, text_encoder.value, tokenizer.value, noise_scheduler.value = load_models(device.value)
44
 
45
+ young = gr.State()
46
+ young.value = get_direction(df, "Young", pinverse, 1000, device.value)
47
  young.value = debias(young.value, "Male", df, pinverse, device.value)
48
  young.value = debias(young.value, "Pointy_Nose", df, pinverse, device.value)
49
  young.value = debias(young.value, "Wavy_Hair", df, pinverse, device.value)
50
  young.value = debias(young.value, "Chubby", df, pinverse, device.value)
51
  young.value = debias(young.value, "No_Beard", df, pinverse, device.value)
52
  young.value = debias(young.value, "Mustache", df, pinverse, device.value)
53
+
54
+ pointy = gr.State()
55
+ pointy,value = get_direction(df, "Pointy_Nose", pinverse, 1000, device.value)
56
  pointy.value = debias(pointy.value, "Young", df, pinverse, device.value)
57
  pointy.value = debias(pointy.value, "Male", df, pinverse, device.value)
58
  pointy.value = debias(pointy.value, "Wavy_Hair", df, pinverse, device.value)
59
  pointy.value = debias(pointy.value, "Chubby", df, pinverse, device.value)
60
  pointy.value = debias(pointy.value, "Heavy_Makeup", df, pinverse, device.value)
61
+
62
+ wavy = gr.State()
63
+ wavy.value = get_direction(df, "Wavy_Hair", pinverse, 1000, device.value)
64
  wavy.value = debias(wavy.value, "Young", df, pinverse, device.value)
65
  wavy.value = debias(wavy.value, "Male", df, pinverse, device.value)
66
  wavy.value = debias(wavy.value, "Pointy_Nose", df, pinverse, device.value)
67
  wavy.value = debias(wavy.value, "Chubby", df, pinverse, device.value)
68
  wavy.value = debias(wavy.value, "Heavy_Makeup", df, pinverse, device.value)
69
 
70
+ thick = gr.State()
71
+ thick.value = get_direction(df, "Bushy_Eyebrows", pinverse, 1000, device.value)
72
  thick.value = debias(thick.value, "Male", df, pinverse, device.value)
73
  thick.value = debias(thick.value, "Young", df, pinverse, device.value)
74
  thick.value = debias(thick.value, "Pointy_Nose", df, pinverse, device.value)