add torchvision to requirement

app.py

@@ -12,19 +12,9 @@ from src.utils_sample import set_seed, resize_and_add_margin
 import os
 
 
-dtype = torch.bfloat16
-token = os.environ.get("HF_TOKEN")
-
-pipe = None
-pipe = FluxPipeline.from_pretrained(
-    "black-forest-labs/FLUX.1-dev", 
-    torch_dtype=dtype, 
-    use_auth_token=token
-)
-pipe = pipe.to("cuda")
 
 @spaces.GPU
-def process_image_and_text(image, scale, seed, text):
+def process_image_and_text(image, scale, seed, text, pipe):
     set_seed(seed)
     print("execution_device 1", pipe._execution_device)
     blended_attn_procs = {}
@@ -147,6 +137,17 @@ header = """
 
 def create_app():
     
+    dtype = torch.bfloat16
+    token = os.environ.get("HF_TOKEN")
+
+    pipe = None
+    pipe = FluxPipeline.from_pretrained(
+        "black-forest-labs/FLUX.1-dev", 
+        torch_dtype=dtype, 
+        use_auth_token=token
+    )
+    pipe = pipe.to("cuda")
+
     with gr.Blocks() as app:
         gr.Markdown(header, elem_id="header")
         with gr.Row(equal_height=False):
@@ -175,10 +176,10 @@ def create_app():
                 inputs=[original_image, scale, seed, text],
                 label="Examples",
             )
-        
+
         submit_btn.click(
             fn=process_image_and_text,
-            inputs=[original_image, scale, seed, text],
+            inputs=[original_image, scale, seed, text, pipe],
             outputs=output_image,
         )