shreyasiv commited on
Commit
6b18d57
1 Parent(s): 47f5142
Files changed (1) hide show
  1. app.py +29 -0
app.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from typing import List
2
+ from pydantic import BaseModel
3
+ from lama_cleaner.server import main
4
+
5
+ class FakeArgs(BaseModel):
6
+ host: str = "0.0.0.0"
7
+ port: int = 7860
8
+ model: str = 'lama'
9
+ hf_access_token: str = ""
10
+ sd_disable_nsfw: bool = False
11
+ sd_cpu_textencoder: bool = True
12
+ sd_run_local: bool = False
13
+ sd_enable_xformers: bool = False
14
+ local_files_only: bool = False
15
+ cpu_offload: bool = False
16
+ device: str = "cpu"
17
+ gui: bool = False
18
+ gui_size: List[int] = [1000, 1000]
19
+ input: str = ''
20
+ disable_model_switch: bool = True
21
+ debug: bool = False
22
+ no_half: bool = False
23
+ disable_nsfw: bool = False
24
+ enable_xformers: bool = False
25
+ model_dir: str = None
26
+ output_dir: str = None
27
+
28
+ if __name__ == "__main__":
29
+ main(FakeArgs())