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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -85,6 +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
  network.value = sample_weights(unet.value, proj.value, mean.value, std.value, v[:, :1000], device.value, factor = 1.00)
89
 
90
  @torch.no_grad()
 
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()