lcccluck's picture
Add QuickDraw diffusion model and app code
9894238 verified
|
Raw
History Blame Contribute Delete
529 Bytes
# QuickDraw Diffusion Web App
Run the local web UI from the repository root:
```bash
uvicorn quickdraw_app.server:app --host 127.0.0.1 --port 7860
```
By default it loads:
```text
models/diffusion/checkpoint_step_500000.pt
```
To use another checkpoint:
```bash
QUICKDRAW_CHECKPOINT=/absolute/path/to/checkpoint.pt uvicorn quickdraw_app.server:app --host 127.0.0.1 --port 7860
```
The recognition button uses the bundled 100-class 64x64 residual CNN:
```text
quickdraw_app/models/cnn_residual_100cls_64/best_model.pt
```