Spaces:
Running
Running
yizhangliu
commited on
Commit
•
a18a2d9
1
Parent(s):
3079cc3
Update app.py
Browse files
app.py
CHANGED
@@ -15,13 +15,14 @@ from torchvision import transforms
|
|
15 |
# from diffusers import DiffusionPipeline
|
16 |
|
17 |
import io
|
|
|
18 |
import multiprocessing
|
19 |
import random
|
20 |
import time
|
21 |
import imghdr
|
22 |
from pathlib import Path
|
23 |
from typing import Union
|
24 |
-
|
25 |
|
26 |
from lama_cleaner.model_manager import ModelManager
|
27 |
from lama_cleaner.schema import Config
|
@@ -55,7 +56,6 @@ if os.environ.get("CACHE_DIR"):
|
|
55 |
|
56 |
BUILD_DIR = os.environ.get("LAMA_CLEANER_BUILD_DIR", "app/build")
|
57 |
|
58 |
-
|
59 |
from share_btn import community_icon_html, loading_icon_html, share_js
|
60 |
|
61 |
HF_TOKEN_SD = os.environ.get('HF_TOKEN_SD')
|
@@ -79,6 +79,8 @@ def process(init_image, mask):
|
|
79 |
origin_image_bytes = input["image"].read()
|
80 |
'''
|
81 |
|
|
|
|
|
82 |
# image, alpha_channel = load_img(origin_image_bytes)
|
83 |
original_shape = init_image.shape
|
84 |
interpolation = cv2.INTER_CUBIC
|
|
|
15 |
# from diffusers import DiffusionPipeline
|
16 |
|
17 |
import io
|
18 |
+
import logging
|
19 |
import multiprocessing
|
20 |
import random
|
21 |
import time
|
22 |
import imghdr
|
23 |
from pathlib import Path
|
24 |
from typing import Union
|
25 |
+
from loguru import logger
|
26 |
|
27 |
from lama_cleaner.model_manager import ModelManager
|
28 |
from lama_cleaner.schema import Config
|
|
|
56 |
|
57 |
BUILD_DIR = os.environ.get("LAMA_CLEANER_BUILD_DIR", "app/build")
|
58 |
|
|
|
59 |
from share_btn import community_icon_html, loading_icon_html, share_js
|
60 |
|
61 |
HF_TOKEN_SD = os.environ.get('HF_TOKEN_SD')
|
|
|
79 |
origin_image_bytes = input["image"].read()
|
80 |
'''
|
81 |
|
82 |
+
print(f'liuyz_1_here_')
|
83 |
+
|
84 |
# image, alpha_channel = load_img(origin_image_bytes)
|
85 |
original_shape = init_image.shape
|
86 |
interpolation = cv2.INTER_CUBIC
|