debugging app.py

app.py

@@ -22,14 +22,14 @@ pipe = FluxPipeline.from_pretrained(
 pipe = pipe.to("cuda")
 
 @spaces.GPU
-def process_image_and_text(image, text, seed, pipeline):
+def process_image_and_text(image, text, seed):
     set_seed(seed)
     # image = Image.open(img_path).convert('RGB') 
     image = resize_and_add_margin(image, target_size=512)
     
     image_list = [image]
-
-    out = pipeline(
+    
+    out = pipe(
         prompt=text,
         height=512,
         width=512,
@@ -154,7 +154,7 @@ def create_app():
             filename="FLUX/it-blender.bin",
             token=token
         )
-        
+        print("pipe._execution_device",pipe._execution_device)
         pretrained_blended_attn_weights = torch.load(model_path, map_location=pipe._execution_device)
 
         key_changed_blended_attn_weights = {}
@@ -173,9 +173,8 @@ def create_app():
                 label="Examples",
             )
 
-        from functools import partial
         submit_btn.click(
-            fn=partial(process_image_and_text, pipeline=pipe),
+            fn=process_image_and_text,
             inputs=[original_image, text, seed],
             outputs=output_image,
         )
@@ -185,4 +184,5 @@ def create_app():
 
 if __name__ == "__main__":
     demo = create_app()
-    demo.launch(debug=True, ssr_mode=False)
+    demo.launch(debug=True, ssr_mode=False, queue=False)
+    # demo.launch(debug=True, ssr_mode=False)