Spaces:
Paused
Paused
Update gradio_app.py
Browse files- gradio_app.py +8 -8
gradio_app.py
CHANGED
@@ -142,13 +142,13 @@ def on_change_model(selected, model):
|
|
142 |
latent, noise = g_ema.prepare([sample_z])
|
143 |
sample, F = g_ema.generate(latent, noise)
|
144 |
|
145 |
-
state =
|
146 |
'latent': latent,
|
147 |
'noise': noise,
|
148 |
'F': F,
|
149 |
'sample': sample,
|
150 |
'history': []
|
151 |
-
}
|
152 |
return model, state, to_image(sample), size
|
153 |
|
154 |
|
@@ -158,14 +158,14 @@ def on_new_image(model):
|
|
158 |
latent, noise = g_ema.prepare([sample_z])
|
159 |
sample, F = g_ema.generate(latent, noise)
|
160 |
|
161 |
-
state =
|
162 |
'latent': latent,
|
163 |
'noise': noise,
|
164 |
'F': F,
|
165 |
'sample': sample,
|
166 |
'history': []
|
167 |
-
}
|
168 |
-
points =
|
169 |
target_point = False
|
170 |
return to_image(sample), to_image(sample), state, points, target_point
|
171 |
|
@@ -213,14 +213,14 @@ def main():
|
|
213 |
|
214 |
""",
|
215 |
)
|
216 |
-
state = gr.
|
217 |
'latent': latent,
|
218 |
'noise': noise,
|
219 |
'F': F,
|
220 |
'sample': sample,
|
221 |
'history': []
|
222 |
})
|
223 |
-
points = {'target': [], 'handle': []}
|
224 |
size = gr.State(1024)
|
225 |
|
226 |
with gr.Row():
|
@@ -272,4 +272,4 @@ def main():
|
|
272 |
if __name__ == '__main__':
|
273 |
import fire
|
274 |
demo = main()
|
275 |
-
fire.Fire(demo.queue(concurrency_count=1, max_size=20).launch)
|
|
|
142 |
latent, noise = g_ema.prepare([sample_z])
|
143 |
sample, F = g_ema.generate(latent, noise)
|
144 |
|
145 |
+
state = {
|
146 |
'latent': latent,
|
147 |
'noise': noise,
|
148 |
'F': F,
|
149 |
'sample': sample,
|
150 |
'history': []
|
151 |
+
}
|
152 |
return model, state, to_image(sample), size
|
153 |
|
154 |
|
|
|
158 |
latent, noise = g_ema.prepare([sample_z])
|
159 |
sample, F = g_ema.generate(latent, noise)
|
160 |
|
161 |
+
state = {
|
162 |
'latent': latent,
|
163 |
'noise': noise,
|
164 |
'F': F,
|
165 |
'sample': sample,
|
166 |
'history': []
|
167 |
+
}
|
168 |
+
points = {'target': [], 'handle': []}
|
169 |
target_point = False
|
170 |
return to_image(sample), to_image(sample), state, points, target_point
|
171 |
|
|
|
213 |
|
214 |
""",
|
215 |
)
|
216 |
+
state = gr.State({
|
217 |
'latent': latent,
|
218 |
'noise': noise,
|
219 |
'F': F,
|
220 |
'sample': sample,
|
221 |
'history': []
|
222 |
})
|
223 |
+
points = gr.State({'target': [], 'handle': []})
|
224 |
size = gr.State(1024)
|
225 |
|
226 |
with gr.Row():
|
|
|
272 |
if __name__ == '__main__':
|
273 |
import fire
|
274 |
demo = main()
|
275 |
+
fire.Fire(demo.queue(concurrency_count=1, max_size=20).launch)
|