Spaces:
Sleeping
Sleeping
SD based Avatar requirements
Browse files- app.py +1 -0
- requirements.txt +2 -0
app.py
CHANGED
|
@@ -10,6 +10,7 @@ import cv2
|
|
| 10 |
import numpy as np
|
| 11 |
from PIL import Image, ImageEnhance, ImageOps
|
| 12 |
from rembg import remove
|
|
|
|
| 13 |
import io
|
| 14 |
|
| 15 |
|
|
|
|
| 10 |
import numpy as np
|
| 11 |
from PIL import Image, ImageEnhance, ImageOps
|
| 12 |
from rembg import remove
|
| 13 |
+
from diffusers import StableDiffusionImg2ImgPipeline
|
| 14 |
import io
|
| 15 |
|
| 16 |
|
requirements.txt
CHANGED
|
@@ -25,6 +25,8 @@ torchvision==0.16.2
|
|
| 25 |
#--extra-index-url https://download.pytorch.org/whl/cu118
|
| 26 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
| 27 |
|
|
|
|
|
|
|
| 28 |
|
| 29 |
|
| 30 |
# Core Dependencies
|
|
|
|
| 25 |
#--extra-index-url https://download.pytorch.org/whl/cu118
|
| 26 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
| 27 |
|
| 28 |
+
# Stable Diffusion
|
| 29 |
+
diffusers>=0.20.0
|
| 30 |
|
| 31 |
|
| 32 |
# Core Dependencies
|