Spaces:
Paused
Paused
Update gradio_app.py
Browse files- gradio_app.py +5 -5
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 = {'target': [], 'handle': []}
|
169 |
target_point = False
|
170 |
return to_image(sample), to_image(sample), state, points, target_point
|
171 |
|
|
|
142 |
latent, noise = g_ema.prepare([sample_z])
|
143 |
sample, F = g_ema.generate(latent, noise)
|
144 |
|
145 |
+
state = gr.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 = gr.State({
|
162 |
'latent': latent,
|
163 |
'noise': noise,
|
164 |
'F': F,
|
165 |
'sample': sample,
|
166 |
'history': []
|
167 |
+
})
|
168 |
+
points = gr.State({'target': [], 'handle': []})
|
169 |
target_point = False
|
170 |
return to_image(sample), to_image(sample), state, points, target_point
|
171 |
|