feng2022 commited on
Commit
fc8f608
1 Parent(s): 69542b0

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("stylegan2-ffhq-config-f","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args, device)
93
  iface = gr.Interface(
94
  fn=predict,
95
  inputs='text',