Update app.py

app.py

@@ -39,7 +39,6 @@ def load_models(device):
     pretrained_model_name_or_path = "stablediffusionapi/realistic-vision-v51" 
     
     revision = None
-    rank = 1
     weight_dtype = torch.bfloat16
     
     # Load scheduler, tokenizer and models.
@@ -318,7 +317,7 @@ class main():
     @spaces.GPU(duration=120)
     def sample_then_run(self):
         self.unet = UNet2DConditionModel.from_pretrained(
-            "stablediffusionapi/realistic-vision-v51" , subfolder="unet", revision=revision
+            "stablediffusionapi/realistic-vision-v51" , subfolder="unet", revision=None
         )
         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)