amildravid4292 commited on
Commit
97cf55a
·
verified ·
1 Parent(s): 84c3929

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -90,7 +90,7 @@ class main():
90
 
91
  def sample_model(self):
92
  self.unet, _, _, _, _ = load_models(self.device)
93
- self.network = self.sample_weights(self.unet, self.proj, self.mean, self.std, self.v[:, :1000], self.device, factor = 1.00)
94
 
95
 
96
  @torch.no_grad()
 
90
 
91
  def sample_model(self):
92
  self.unet, _, _, _, _ = load_models(self.device)
93
+ self.network = sample_weights(self.unet, self.proj, self.mean, self.std, self.v[:, :1000], self.device, factor = 1.00)
94
 
95
 
96
  @torch.no_grad()