add torchvision to requirement

app.py

@@ -14,7 +14,7 @@ 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, 
@@ -24,7 +24,7 @@ pipe = pipe.to("cuda")
 
 
 @spaces.GPU
-def process_image_and_text(image, scale, seed, text, pipe):
+def process_image_and_text(image, scale, seed, text):
     set_seed(seed)
     blended_attn_procs = {}
     for name, _ in pipe.transformer.attn_processors.items():
@@ -52,9 +52,8 @@ def process_image_and_text(image, scale, seed, text, pipe):
         key_changed_blended_attn_weights[changed_key] = value.to(dtype)
         
     missing_keys, unexpected_keys = pipe.transformer.load_state_dict(key_changed_blended_attn_weights, strict=False)
-
     pipe = pipe.to("cuda")
-
+    
     # image = Image.open(img_path).convert('RGB') 
     image = resize_and_add_margin(image, target_size=512)
     
@@ -145,7 +144,7 @@ header = """
 """
 
 
-def create_app(pipe):
+def create_app():
     
     with gr.Blocks() as app:
         gr.Markdown(header, elem_id="header")
@@ -178,7 +177,7 @@ def create_app(pipe):
         
         submit_btn.click(
             fn=process_image_and_text,
-            inputs=[original_image, scale, seed, text, pipe],
+            inputs=[original_image, scale, seed, text],
             outputs=output_image,
         )
         
@@ -186,4 +185,4 @@ def create_app(pipe):
 
 
 if __name__ == "__main__":
-    create_app(pipe).launch(debug=True, ssr_mode=False)
+    create_app().launch(debug=True, ssr_mode=False)