YuxinJ commited on
Commit
55e3abb
·
verified ·
1 Parent(s): 3cfbf8d

fix gradio upgrade decrepancy

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -136,20 +136,20 @@ def main():
136
  global model
137
  model = initialize()
138
 
139
- gr.Interface(
140
  inference,
141
  gr.Image(type="pil", label='Input'),
142
- gr.Image(type="pil", label='Output').style(height=300),
143
  theme=args.theme,
144
  title=TITLE,
145
  description=DESCRIPTION,
146
  article=ARTICLE,
147
  examples=EXAMPLES,
148
- allow_screenshot=args.allow_screenshot,
149
- allow_flagging=args.allow_flagging,
150
  live=args.live
151
- ).launch(
152
- enable_queue=args.enable_queue,
 
153
  server_port=args.port,
154
  share=args.share
155
  )
 
136
  global model
137
  model = initialize()
138
 
139
+ demo = gr.Interface(
140
  inference,
141
  gr.Image(type="pil", label='Input'),
142
+ gr.Image(type="pil", label='Output', height=300),
143
  theme=args.theme,
144
  title=TITLE,
145
  description=DESCRIPTION,
146
  article=ARTICLE,
147
  examples=EXAMPLES,
148
+ flagging_mode=args.allow_flagging,
 
149
  live=args.live
150
+ )
151
+ demo.queue(max_size=20)
152
+ demo.launch(
153
  server_port=args.port,
154
  share=args.share
155
  )