fffiloni commited on
Commit
4d45bdc
1 Parent(s): fbe7fdb

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +1 -1
gradio_app.py CHANGED
@@ -192,7 +192,7 @@ def main():
192
 
193
  with gr.Blocks() as demo:
194
  wrapped_model = ModelWrapper()
195
- model = wrapped_model
196
  sample_z = torch.randn([1, 512], device=device)
197
  latent, noise = wrapped_model.g_ema.prepare([sample_z])
198
  sample, F = wrapped_model.g_ema.generate(latent, noise)
 
192
 
193
  with gr.Blocks() as demo:
194
  wrapped_model = ModelWrapper()
195
+ model = gr.State(wrapped_model)
196
  sample_z = torch.randn([1, 512], device=device)
197
  latent, noise = wrapped_model.g_ema.prepare([sample_z])
198
  sample, F = wrapped_model.g_ema.generate(latent, noise)