feng2022 commited on
Commit
1bf16a2
1 Parent(s): 3c13c0a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -89,7 +89,7 @@ def main():
89
  input='text',
90
  ))
91
  device = th.device()
92
- #generator = create_generator(args, device)
93
  iface = gr.Interface(
94
  fn=predict,
95
  inputs='text',
 
89
  input='text',
90
  ))
91
  device = th.device()
92
+ generator = create_generator(args, device)
93
  iface = gr.Interface(
94
  fn=predict,
95
  inputs='text',