Create lehenga_generator.py

lehenga_generator.py

@@ -0,0 +1,36 @@
+import streamlit as st
+from PIL import Image
+from diffusers import DiffusionPipeline
+
+# Load the model and LoRA weights
+@st.cache_resource
+def load_model():
+    pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
+    pipe.load_lora_weights("tryonlabs/FLUX.1-dev-LoRA-Lehenga-Generator")
+    return pipe
+
+pipe = load_model()
+
+# App interface
+st.title("Lehenga Dress Image Generator")
+st.write("Enter a description to generate an image of a lehenga dress.")
+
+# Input prompt
+prompt = st.text_area("Enter your prompt:", 
+                      "A flat-lay image of a lehenga with a traditional style and a fitted waistline is elegantly crafted from stretchy silk material, ensuring a comfortable and flattering fit. The long hemline adds a touch of grace and sophistication to the ensemble. Adorned in a solid blue color, it features a sleeveless design that complements its sweetheart neckline. The solid pattern and the luxurious silk fabric together create a timeless and chic look that is perfect for special occasions.")
+
+# Generate button
+if st.button("Generate Image"):
+    if prompt.strip():
+        with st.spinner("Generating image..."):
+            try:
+                # Generate the image
+                result = pipe(prompt).images[0]
+                # Display the image
+                st.image(result, caption="Generated Lehenga Image", use_column_width=True)
+            except Exception as e:
+                st.error(f"An error occurred: {e}")
+    else:
+        st.warning("Please enter a valid prompt.")
+
+st.write("This app uses AI to generate images of lehenga dresses based on your input description.")