├── HUGGING_FACE_ALL.ipynb
└── README.md


/README.md:
--------------------------------------------------------------------------------
 1 | # 🚀 AI Image Generation & Object Detection App
 2 | 
 3 | Welcome to the **AI Image Generation & Object Detection App**! This project utilizes **Stable Diffusion** for text-to-image generation and **DETR (DEtection TRansformer)** for object detection in images. 🎨🔍
 4 | 
 5 | ---
 6 | 
 7 | ## ✨ Features
 8 | 
 9 | ✅ **Generate Images** from text prompts using Stable Diffusion 🖼️  
10 | ✅ **Detect Objects** in uploaded images with DETR 🤖  
11 | ✅ **Download Generated Images** with a single click ⬇️  
12 | ✅ **Streamlit Web Interface** for an easy-to-use experience 🏆  
13 | 
14 | ---
15 | 
16 | ## 📦 Installation
17 | 
18 | 1. Clone this repository:
19 |    ```bash
20 |    https://github.com/machinelearningprodigy/objectDetection
21 |    
22 |    ```
23 |     ```bash
24 |     cd objectDetection
25 |       ```
26 | 
27 | 2. Install dependencies:
28 |    ```bash
29 |    pip install -r requirements.txt
30 |    ```
31 | 3. Download the Stable Diffusion model weights:
32 |    ```bash
33 |    mkdir stable_diffusion_weights
34 |    ```
35 |    Place your model weights inside the `stable_diffusion_weights` folder.
36 | 
37 | 4. Run the app:
38 |    ```bash
39 |    streamlit run app.py
40 |    ```
41 | 
42 | ---
43 | 
44 | ## 🎨 Image Generation
45 | 
46 | 1️⃣ Enter a **text prompt** (e.g., *"A futuristic city at sunset"*)  
47 | 2️⃣ Click **Generate Image**  
48 | 3️⃣ View and **download** your AI-generated masterpiece! 🌟
49 | 
50 | ---
51 | 
52 | ## 🔍 Object Detection
53 | 
54 | 1️⃣ Upload an **image**  
55 | 2️⃣ The model will **detect objects** and display bounding boxes 🖼️  
56 | 3️⃣ View detected objects with labels and confidence scores 📊
57 | 
58 | ---
59 | 
60 | ## 🛠️ Technologies Used
61 | 
62 | - 🤖 **Stable Diffusion** (RunwayML Stable Diffusion v1-5)
63 | - 🔍 **DETR (DEtection TRansformer)**
64 | - 🚀 **Streamlit** (Interactive UI)
65 | - 🏞️ **PIL (Pillow)** (Image Processing)
66 | - 🔥 **PyTorch** (Deep Learning Framework)
67 | 
68 | ---
69 | 
70 | ## 🌟 Demo
71 | 
72 | <p align="center">
73 |   <img src="https://media.giphy.com/media/3ohs7KViF9UczbUiDO/giphy.gif" width="400">
74 |   <br>
75 |   <em>Watch the AI generate images in real-time! 🎨</em>
76 | </p>
77 | 
78 | ---
79 | 
80 | ## 📜 License
81 | 
82 | This project is licensed under the **MIT License**. Feel free to use and modify it! 💡
83 | 
84 | 💙 **Enjoy AI Creativity!** 🚀
85 | 
86 | 


--------------------------------------------------------------------------------