Update app.py
Browse files
app.py
CHANGED
|
@@ -10,6 +10,7 @@ import imageio
|
|
| 10 |
from torchvision.utils import save_image
|
| 11 |
from PIL import Image
|
| 12 |
import io
|
|
|
|
| 13 |
from transformers import AutoModel, AutoConfig
|
| 14 |
from rembg import remove, new_session
|
| 15 |
from functools import partial
|
|
|
|
| 10 |
from torchvision.utils import save_image
|
| 11 |
from PIL import Image
|
| 12 |
import io
|
| 13 |
+
from io import BytesIO
|
| 14 |
from transformers import AutoModel, AutoConfig
|
| 15 |
from rembg import remove, new_session
|
| 16 |
from functools import partial
|