jhj0517 commited on
Commit
64bd1c5
1 Parent(s): 5db951d

Add model dir & output dir customization

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -14,7 +14,10 @@ class App:
14
  self.args = args
15
  self.app = gr.Blocks(css=GRADIO_CSS)
16
  self.i18n = Translate(I18N_YAML_PATH)
17
- self.inferencer = LivePortraitInferencer()
 
 
 
18
 
19
  @staticmethod
20
  def create_parameters():
 
14
  self.args = args
15
  self.app = gr.Blocks(css=GRADIO_CSS)
16
  self.i18n = Translate(I18N_YAML_PATH)
17
+ self.inferencer = LivePortraitInferencer(
18
+ model_dir=args.model_dir if args else MODELS_DIR,
19
+ output_dir=args.output_dir if args else OUTPUTS_DIR
20
+ )
21
 
22
  @staticmethod
23
  def create_parameters():