PKUWilliamYang commited on
Commit
8747810
1 Parent(s): acc813f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py CHANGED
@@ -9,6 +9,17 @@ import gradio as gr
9
 
10
  from vtoonify_model import Model
11
 
 
 
 
 
 
 
 
 
 
 
 
12
  DESCRIPTION = '''
13
  <div align=center>
14
  <h1 style="font-weight: 900; margin-bottom: 7px;">
 
9
 
10
  from vtoonify_model import Model
11
 
12
+ def parse_args() -> argparse.Namespace:
13
+ parser = argparse.ArgumentParser()
14
+ parser.add_argument('--device', type=str, default='cpu')
15
+ parser.add_argument('--theme', type=str)
16
+ parser.add_argument('--share', action='store_true')
17
+ parser.add_argument('--port', type=int)
18
+ parser.add_argument('--disable-queue',
19
+ dest='enable_queue',
20
+ action='store_false')
21
+ return parser.parse_args()
22
+
23
  DESCRIPTION = '''
24
  <div align=center>
25
  <h1 style="font-weight: 900; margin-bottom: 7px;">