Update app.py

app.py

@@ -4,22 +4,18 @@ from PIL import Image
 from io import BytesIO
 import os
 import requests
+from diffusers import DiffusionPipeline
 
 st.sidebar.title("Stable Diffusion SDXL-Turbo", help="Made with API")
 st.sidebar.subheader("By KVI Kontent")
 st.sidebar.write("Choose model and enter prompt")
-api_key = os.environ['api_key']
-API_URL_DALLE = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
-headers = {"Authorization": f"Bearer {api_key}"}
+pipeline = DiffusionPipeline.from_pretrained("stablediffusionapi/juggernaut-xl-v5")
+pipeline.load_lora_weights("openskyml/dalle-3-xl")
 
 model = st.sidebar.selectbox("Choose Model", ("prodia", "pollinations", "Dall-e 3"))
 
 prompt = st.sidebar.text_input("Prompt", "")
 
-def Dalle_query(payload):
-    response = requests.post(API_URL_DALLE, headers=headers, json=payload)
-    return response.content
-
 if model == "prodia" or model == "pollinations":
     try:
         resp = Client.create_generation(model, prompt)
@@ -28,8 +24,6 @@ if model == "prodia" or model == "pollinations":
     except Exception as e:
         st.error(str(e))
 elif model == "Dall-e 3":
-    image_bytes = Dalle_query({
-    	"inputs": prompt
-    })
+    image_bytes = pipeline(prompt)
     image = Image.open(BytesIO(image_bytes))
     st.image(image, caption="Generated Image")