Sanster's picture
Update app.py
0a62c57
from typing import List
from pydantic import BaseModel
from lama_cleaner.server import main
class FakeArgs(BaseModel):
host: str = "0.0.0.0"
port: int = 7860
model: str = 'lama'
hf_access_token: str = ""
sd_enable_xformers: bool = False
sd_disable_nsfw: bool = False
sd_cpu_textencoder: bool = True
sd_controlnet: bool = False
sd_controlnet_method: str = "control_v11p_sd15_canny"
sd_local_model_path: str = ""
sd_run_local: bool = False
local_files_only: bool = False
cpu_offload: bool = False
device: str = "cpu"
gui: bool = False
gui_size: List[int] = [1000, 1000]
input: str = ''
disable_model_switch: bool = True
debug: bool = False
no_half: bool = False
disable_nsfw: bool = False
enable_xformers: bool = False
enable_interactive_seg: bool = True
interactive_seg_model: str = "vit_b"
interactive_seg_device: str = "cpu"
enable_remove_bg: bool = False
enable_anime_seg: bool = False
enable_realesrgan: bool = False
enable_gfpgan: bool = False
gfpgan_device: str = "cpu"
enable_restoreformer: bool = False
enable_gif: bool = False
quality: int = 95
model_dir: str = None
output_dir: str = None
if __name__ == "__main__":
main(FakeArgs())