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

Update app.py

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