| # Diffusion Models App | |
| A Python application that uses Hugging Face inference endpoints for text-to-image and image-to-image generation with a Gradio UI and API endpoints. | |
| ## Features | |
| - Text-to-image generation | |
| - Image-to-image transformation with optional prompt | |
| - Gradio UI for interactive use | |
| - API endpoints for integration with other applications | |
| - Configurable models via text input | |
| - Default values for prompts, negative prompts, and models | |
| ## Project Structure | |
| - `main.py` - Entry point that can run both UI and API | |
| - `app.py` - Gradio UI implementation | |
| - `api.py` - FastAPI server for API endpoints | |
| - `inference.py` - Core functionality for HF inference | |
| - `config.py` - Configuration and settings | |
| - `requirements.txt` - Dependencies | |
| ## Setup & Usage | |
| 1. Clone the repository | |
| 2. Create a .env file with your Hugging Face token (copy from .env.example) | |
| 3. Install dependencies: `pip install -r requirements.txt` | |
| 4. Run the application: `python main.py` | |
| ## Running Options | |
| - Run both UI and API: `python main.py` | |
| - Run only the API: `python main.py --mode api` | |
| - Run only the UI: `python main.py --mode ui` | |
| ## API Endpoints | |
| - `POST /text-to-image` - Generate an image from text | |
| - `POST /image-to-image` - Transform an image with optional prompt | |
| ## Default Values | |
| The application includes defaults for: | |
| - Sample prompts for text-to-image and image-to-image | |
| - Negative prompts to exclude unwanted elements | |
| - Pre-filled model names for both text-to-image and image-to-image | |
| These defaults are applied to both the Gradio UI and API endpoints for consistency. | |
| ## Environment Variables | |
| - `HF_TOKEN` - Your Hugging Face API token | |
| - `API_HOST` - Host for the API server (default: 0.0.0.0) | |
| - `API_PORT` - Port for the API server (default: 8000) | |
| - `GRADIO_HOST` - Host for the Gradio UI (default: 0.0.0.0) | |
| - `GRADIO_PORT` - Port for the Gradio UI (default: 7860) | |