Update app.py

app.py

@@ -83,6 +83,7 @@ thick.value = debias(thick.value, "Brown_Hair", df, pinverse, device.value)
 thick.value = debias(thick.value, "Pale_Skin", df, pinverse, device.value)
 thick.value = debias(thick.value, "Heavy_Makeup", df, pinverse, device.value)
 
+@torch.no_grad()
 def sample_model():
     unet.value, _, _, _, _ = load_models(device.value)
     network.value = sample_weights(unet.value, proj, mean, std, v[:, :1000], device.value, factor = 1.00)
@@ -207,7 +208,8 @@ def edit_inference(prompt, negative_prompt, guidance_scale, ddim_steps, seed, st
     network.value.reset()
 
     return image
-
+    
+@torch.no_grad()
 @spaces.GPU
 def sample_then_run():
     sample_model()