Update app.py

app.py

@@ -48,6 +48,9 @@ def load_pipeline():
     return editor_pipeline
 
 
+pipeline = load_pipeline()
+
+
 def set_env(seed=0):
     torch.manual_seed(seed)
     torch.set_grad_enabled(False)
@@ -67,11 +70,8 @@ def generate_img(
     scale,
     image_guidance_scale,
     seed,
-    pipeline: ImageEditor,
+    progress=gr.Progress(track_tqdm=True),
 ):
-    # FIXME
-    pipeline = load_pipeline()
-    
     # pipeline.pipe.to("cuda").to(pipeline.weight_dtype)
     
     pil_image = PILImage.fromarray(np_image)
@@ -96,14 +96,13 @@ if __name__ == "__main__":
     image_dir = pathlib.Path('images')
     examples = [[path.as_posix(), "let this case swim in the river", 20, 4.5, 1.2, 42] for path in sorted(image_dir.glob('*.png'))]
 
-    editor_pipeline = load_pipeline()
-    generate_fn = partial(
-        generate_img,
-        pipeline=editor_pipeline
-    )
+    # generate_fn = partial(
+    #     generate_img,
+    #     pipeline=editor_pipeline
+    # )
 
     demo = gr.Interface(
-        fn=generate_fn,
+        fn=generate_img,
         inputs=[
             gr.Image(label="Input", type="numpy"),
             Textbox(label="Prompt", placeholder="Please enter your prompt. \n"),