Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
-
|
| 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
|