Update app.py

app.py

@@ -26,7 +26,7 @@ def load_beat_model():
 # @st.cache_resource
 def load_image_model():
      torch.cuda.empty_cache()
-     pipeline = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, revision="fp16")
+     pipeline = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", variant='fp16')
      pipeline.to("cuda")
      pipeline.load_lora_weights("Weights/pytorch_lora_weights.safetensors", weight_name="pytorch_lora_weights.safetensors")
      return pipeline