Update app.py

app.py

@@ -45,9 +45,7 @@ def resize_image(input_path, output_path, target_height):
 
 @spaces.GPU(duration=90)
 def infer(image_in, prompt, inference_steps, guidance_scale, control_weight, progress=gr.Progress(track_tqdm=True)):
-    pipe = FluxControlNetPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.bfloat16)
-    # pipe.enable_model_cpu_offload()
-    pipe.to("cuda")
+    
 
     n_prompt = 'NSFW, nude, naked, porn, ugly'
 
@@ -64,7 +62,7 @@ def infer(image_in, prompt, inference_steps, guidance_scale, control_weight, pro
     # infer
     image = pipe(
         prompt=prompt,
-        negative_prompt=n_prompt,
+        #negative_prompt=n_prompt,
         control_image=control_image, 
         controlnet_conditioning_scale=control_weight,
         num_inference_steps=inference_steps,