move pipe

app.py

@@ -2,14 +2,20 @@ import spaces
 import torch
 from diffusers import FluxInpaintPipeline
 
+dtype = torch.bfloat16
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+pipe = FluxInpaintPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16).to(device)
+
+
+
 import gradio as gr
 import re
 from PIL import Image
-import flux1_inpaint
+
 import os
 import numpy as np
-import shutil
-#shutil.rmtree("/home/user/app/.gradio/cached_examples/23")
+
 
 def sanitize_prompt(prompt):
   # Allow only alphanumeric characters, spaces, and basic punctuation
@@ -43,9 +49,7 @@ def process_images(image, image2=None,prompt="a girl",inpaint_model="black-fores
     def process_image(image,mask_image,prompt="a person",model_id="black-forest-labs/FLUX.1-schnell",strength=0.75,seed=0,num_inference_steps=4):
         if image == None:
             return None
-        
-        pipe = FluxInpaintPipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16)
-        pipe.to("cuda")
+   
 
         generators = []
         generator = torch.Generator("cuda").manual_seed(seed)