Update app.py

app.py

@@ -84,6 +84,7 @@ 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()
+@spaces.GPU
 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)