Update gradio_app.py

gradio_app.py

@@ -142,13 +142,13 @@ def on_change_model(selected, model):
     latent, noise = g_ema.prepare([sample_z])
     sample, F = g_ema.generate(latent, noise)
 
-    state = gr.State({
+    state = {
         'latent': latent,
         'noise': noise,
         'F': F,
         'sample': sample,
         'history': []
-    })
+    }
     return model, state, to_image(sample), size
 
 
@@ -158,14 +158,14 @@ def on_new_image(model):
     latent, noise = g_ema.prepare([sample_z])
     sample, F = g_ema.generate(latent, noise)
 
-    state = gr.State({
+    state = {
         'latent': latent,
         'noise': noise,
         'F': F,
         'sample': sample,
         'history': []
-    })
-    points = gr.State({'target': [], 'handle': []})
+    }
+    points = {'target': [], 'handle': []}
     target_point = False
     return to_image(sample), to_image(sample), state, points, target_point
 
@@ -213,14 +213,14 @@ def main():
             
             """,
         )
-        state = gr.Sate({
+        state = gr.State({
             'latent': latent,
             'noise': noise,
             'F': F,
             'sample': sample,
             'history': []
         })
-        points = {'target': [], 'handle': []}
+        points = gr.State({'target': [], 'handle': []})
         size = gr.State(1024)
 
         with gr.Row():
@@ -272,4 +272,4 @@ def main():
 if __name__ == '__main__':
     import fire
     demo = main()
-    fire.Fire(demo.queue(concurrency_count=1, max_size=20).launch)
+    fire.Fire(demo.queue(concurrency_count=1, max_size=20).launch)