yizhangliu commited on
Commit
a18a2d9
1 Parent(s): 3079cc3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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
- # from loguru import logger
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