This web app generate image using Stable Diffusion v2.1 from Stability AI. The web app use FastAPI for the backend and STreamlit for the front end The model is obtained from Huggingface Stable Diffusion v2-1 Model Card Please comply with the term of use as indicated on the model card page when using the web app
-
Git clone the repo
git clone https://github.com/haizadtarik/ai-image-generation.git
-
Install Dependencies
pip install -r requirements.txt
-
Bring up the backend server
python server.py
-
Bring up the web app
streamlit run app.py
or if the backend server is running on different system
streamlit run app.py -- <BACKEND_URL>
```
docker build -t <NAME> .
docker run -p 8080:8080 -p 8501:8501 <NAME>
```