Init commit

README.md

@@ -1,13 +1,44 @@
----
-title: Blog Post
-emoji: 📊
-colorFrom: purple
-colorTo: blue
-sdk: gradio
-sdk_version: 4.39.0
-app_file: app.py
-pinned: false
-license: mit
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Blog Post Generator
+
+This is a web application that generates complete blog posts, including titles, content, cover images, and summaries based on user input. The app is built using Gradio and is designed to be deployed on Hugging Face Spaces.
+
+## Features
+
+- Generate blog post title using a specialized title generation model
+- Generate blog post content using GPT-2
+- Create a cover image for the blog post using Stable Diffusion
+- Summarize the generated article using BART
+- Simple and intuitive user interface
+
+## Models Used
+
+- Title Generation: fabiochiu/t5-small-medium-title-generation
+- Text Generation: gpt2
+- Image Generation: runwayml/stable-diffusion-v1-5
+- Summarization: facebook/bart-large-cnn
+
+## Setup and Deployment
+
+1. Fork this repository
+2. Set up a new Hugging Face Space (make sure to use a GPU-enabled instance)
+3. Configure the GitHub Actions workflow by adding your Hugging Face token as a secret named `HF_TOKEN`
+4. Update the `deploy.yml` file with your Hugging Face username and space name
+5. Push changes to the `main` branch to trigger automatic deployment
+
+## Local Development
+
+To run the app locally:
+
+1. Clone the repository
+2. Install the required packages: `pip install -r requirements.txt`
+3. Run the app: `python app.py`
+
+Note: Running multiple large language models and Stable Diffusion locally requires significant computational resources. A GPU is strongly recommended for reasonable performance.
+
+## Contributing
+
+Feel free to open issues or submit pull requests to improve the application.
+
+## License
+
+This project is open source and available under the [MIT License](LICENSE).

app.py

@@ -0,0 +1,48 @@
+import gradio as gr
+from transformers import pipeline
+from diffusers import StableDiffusionPipeline
+import torch
+
+# Initialize the pipelines
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+text_generator = pipeline("text-generation", model="gpt2", device=device)
+summarizer = pipeline("summarization", model="facebook/bart-large-cnn", device=device)
+title_generator = pipeline("text2text-generation", model="fabiochiu/t5-small-medium-title-generation", device=device)
+
+# Initialize the Stable Diffusion pipeline
+stable_diffusion = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
+stable_diffusion.to(device)
+
+# Function to generate blog post
+def generate_blog_post(query):
+    # Generate the article
+    article = text_generator(query, max_length=500, num_return_sequences=1)[0]['generated_text']
+    
+    # Generate a title for the article
+    title = title_generator(article, max_length=30, num_return_sequences=1)[0]['generated_text']
+    
+    # Generate a cover image using Stable Diffusion
+    cover_image = stable_diffusion(query, num_inference_steps=50, guidance_scale=7.5).images[0]
+    
+    # Generate a summary of the article
+    summary = summarizer(article, max_length=100, min_length=30, do_sample=False)[0]['summary_text']
+    
+    return title, article, cover_image, summary
+
+# Create the Gradio interface
+iface = gr.Interface(
+    fn=generate_blog_post,
+    inputs=gr.Textbox(lines=2, placeholder="Enter your blog post topic..."),
+    outputs=[
+        gr.Textbox(label="Generated Title"),
+        gr.Textbox(label="Generated Article"),
+        gr.Image(label="Cover Image"),
+        gr.Textbox(label="Article Summary")
+    ],
+    title="Blog Post Generator",
+    description="Enter a topic, and I'll generate a blog post with a title, cover image, and summary!"
+)
+
+# Launch the app
+iface.launch()

requirements.txt

@@ -0,0 +1,6 @@
+gradio==3.23.0
+transformers==4.28.1
+torch==1.13.1
+diffusers==0.14.0
+accelerate==0.18.0
+sentencepiece==0.1.97