Update app.py

app.py

@@ -8,25 +8,14 @@ from flask import Flask, render_template, request, url_for
 
 app = Flask(__name__)
 
-model_id = "CompVis/stable-diffusion-v1-4"
+model_id = "runwayml/stable-diffusion-v1-5"
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 #pipe = StableDiffusionPipeline.from_pretrained(model_id)
 #pipe = pipe.to(device)
 #pipe.enable_attention_slicing()
 
-pipe = StableDiffusionXLPipeline.from_pretrained(
-        "fluently/Fluently-XL-v2",
-        torch_dtype=torch.float16,
-        use_safetensors=True,
-    )
-
-pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-
-pipe.load_lora_weights("ehristoforu/dalle-3-xl-v2")
-pipe.set_adapters("dalle")
-
-pipe = pipe.to(device)
+pipe = StableDiffusionPipeline.from_pretrained(model_id)
 
 @app.route('/', methods=['GET', 'POST'])
 def generate_image():