Update app.py

app.py

@@ -28,7 +28,7 @@ import diffusers
 
 # init
 dtype = torch.bfloat16
-device = "cuda" if torch.cuda.is_available() else "cpu"
+device = "cuda"
 base_model = "black-forest-labs/FLUX.1-dev"
 
 # load pipe
@@ -71,11 +71,7 @@ def generate_image(orginal_image, prompt, adapter_names,  steps, seed, image_str
     
     gr.Info("Start to generate images ...")
     with calculateDuration(f"Make a new generator: {seed}"):
-        # if orginal_image:
-        #     img2img_pipe.to(device)
-        # else:
-        #     txt2img_pipe.to(device)
-        
+        txt2img_pipe.to(device)  
         generator = torch.Generator(device=device).manual_seed(seed)
     
     with calculateDuration("Generating image"):