Update app.py

app.py

@@ -317,7 +317,7 @@ class main():
     @spaces.GPU(duration=120)
     def sample_then_run(self):
         self.unet = UNet2DConditionModel.from_pretrained(
-            pretrained_model_name_or_path, subfolder="unet", revision=revision
+            "stablediffusionapi/realistic-vision-v51" , subfolder="unet", revision=revision
         )
         self.unet.to(self.device, dtype=torch.bfloat16)
         self.weights = sample_weights(self.unet, self.proj, self.mean, self.std, self.v[:, :1000], self.device, factor = 1.00)