Spaces:
Paused
Paused
Update gradio_app.py
Browse files- 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)
|