Update app.py

app.py

@@ -10,6 +10,9 @@ class FakeArgs(BaseModel):
     sd_disable_nsfw: bool = False
     sd_cpu_textencoder: bool = True
     sd_run_local: bool = False
+    sd_enable_xformers: bool = False
+    local_files_only: bool = False
+    cpu_offload: bool = False
     device: str = "cpu"
     gui: bool = False
     gui_size: List[int] = [1000, 1000]
@@ -17,6 +20,10 @@ class FakeArgs(BaseModel):
     disable_model_switch: bool = True
     debug: bool = False
     no_half: bool = False
+    disable_nsfw: bool = False
+    enable_xformers: bool = False
+    model_dir: str = None
+    output_dir: str = None
 
 if __name__ == "__main__":
     main(FakeArgs())