amildravid4292 commited on
Commit
c921f7a
·
verified ·
1 Parent(s): cf1742e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -85,8 +85,7 @@ thick.value = debias(thick.value, "Heavy_Makeup", df, pinverse, device.value)
85
 
86
  def sample_model():
87
  unet.value, _, _, _, _ = load_models(device.value)
88
- print(network)
89
- network.value = sample_weights(unet.value, proj.value, mean.value, std.value, v[:, :1000], device.value, factor = 1.00)
90
 
91
  @torch.no_grad()
92
  @spaces.GPU
 
85
 
86
  def sample_model():
87
  unet.value, _, _, _, _ = load_models(device.value)
88
+ network.value = sample_weights(unet.value, proj, mean, std, v[:, :1000], device.value, factor = 1.00)
 
89
 
90
  @torch.no_grad()
91
  @spaces.GPU