Update app.py

app.py

@@ -1,36 +1,15 @@
 import streamlit as st
-import requests
-from PIL import Image
-from io import BytesIO
+from diffusers import DiffusionPipeline
 
-st.title('CreativeAI')
+pipeline = DiffusionPipeline.from_pretrained("dataautogpt3/OpenDalleV1.1")
 
-# Description
-st.write("""
-This app uses HuggingFace to generate an image based on the provided prompt. You can either select an example prompt from the dropdown list or enter your own prompt.
-Click the "Generate Image" button to initiate the image generation process. Based on KVIImager API by KVI Kontent.
-""")
+prompt = st.text_area("Enter your prompt here:")
 
-# Pre-defined prompts
-example_prompts = ["A cat sitting on a keyboard", "A futuristic cityscape", "Magical forest at night"]
+button = st.button("Generate Image")
 
-# Select box for choosing examples
-selected_prompt = st.selectbox("Select an example prompt", example_prompts)
-
-# Input field for custom prompt
-custom_prompt = st.text_input("Enter your own prompt")
-
-prompt = selected_prompt if not custom_prompt else custom_prompt
-
-if st.button('Generate Image'):
-    if prompt:
-        with st.spinner('Generating your image...'):
-            url = 'https://57b7-217-71-237-228.ngrok-free.app/generate_image'
-            params = {'prompt': prompt}
-            response = requests.get(url, params=params)
-
-            if response.status_code == 200:
-                image = Image.open(BytesIO(response.content))
-                st.image(image, caption='Generated Image', use_column_width=True)
-            else:
-                st.error("Failed to generate the image. Please try again.")
+if button:
+    if prompt != '':
+        image = pipeline.generate(prompt=prompt)
+        st.image(image, caption=f"Generated image from prompt: {prompt}")
+    else:
+        st.error("Please enter a prompt before generating an image.")