Create app.py

app.py

@@ -0,0 +1,80 @@
+import torch
+from diffusers import FluxPipeline
+import streamlit as st
+
+pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
+pipe.enable_model_cpu_offload()  # Save VRAM by offloading to CPU (remove if you have enough GPU power)
+
+st.set_page_config(page_title="🔥 MemeForge - Text-to-Meme Generator 🔥", page_icon="🔥", layout="wide")
+st.markdown("""
+    <style>
+        .title {
+            font-size: 3em;
+            font-weight: bold;
+            color: #FF6347;
+            text-align: center;
+            padding: 20px;
+        }
+        .description {
+            font-size: 1.2em;
+            text-align: center;
+            color: #666;
+        }
+        .input-box {
+            font-size: 1.5em;
+            padding: 10px;
+            border-radius: 8px;
+            border: 2px solid #FF6347;
+            margin: 10px auto;
+            display: block;
+            width: 60%;
+            background-color: #fff3e3;
+        }
+        .generate-button {
+            background-color: #FF6347;
+            color: white;
+            font-size: 1.2em;
+            padding: 10px 30px;
+            border: none;
+            border-radius: 8px;
+            cursor: pointer;
+        }
+        .generate-button:hover {
+            background-color: #FF4500;
+        }
+        .footer {
+            text-align: center;
+            font-size: 1.2em;
+            color: #888;
+            padding: 30px;
+        }
+    </style>
+""", unsafe_allow_html=True)
+
+st.markdown('<div class="title">🔥 MemeForge - Text-to-Meme Generator 🔥</div>', unsafe_allow_html=True)
+st.markdown('<div class="description">Enter a description, and the AI will generate a meme image for you!</div>', unsafe_allow_html=True)
+input_text = st.text_input("Enter meme description:", key="input", placeholder="e.g., A funny dog looking confused.")
+if st.button("Generate Meme", key="generate", help="Click to generate a meme from your description"):
+    if input_text:
+        modified_prompt = "MEME of an Image of " + input_text
+
+        image = pipe(
+            modified_prompt,
+            height=1024,
+            width=1024,
+            guidance_scale=3.5,
+            num_inference_steps=50,
+            max_sequence_length=512,
+            generator=torch.Generator("cpu").manual_seed(0)
+        ).images[0]
+
+        st.image(image, caption="Your Generated Meme", use_column_width=True)
+
+        # Save the image
+        image.save("generated_meme.png")
+    else:
+        st.warning("Please enter a description to generate a meme.")
+
+
+
+