Update app.py

app.py

@@ -11,19 +11,11 @@ openpose = OpenposeDetector.from_pretrained('lllyasviel/ControlNet')
 controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-openpose", torch_dtype=torch.float16)
 pipe = StableDiffusionControlNetPipeline.from_pretrained("helkoo/jelaba_2HR", controlnet=controlnet, safety_checker=None, torch_dtype=torch.float16)
 
+
 #optimizations
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
-#pipe.enable_xformers_memory_efficient_attention()
-#pipe.enable_model_cpu_offload()
-pipe = pipe.to("cuda")
+pipe = pipe.to("cpu")
 
-def generate(image,prompt):
-    image = openpose(image)
-    #image = image
-    image = pipe(prompt, image, num_inference_steps=20).images[0]
-    return image
-
-gr.Interface(fn=generate, inputs=["image","text"], outputs="image").launch(share=True, debug=True)
 
 import numpy as np
 import requests
@@ -49,4 +41,3 @@ gr.Interface(fn=generate2, inputs=["text",
                                   ["S", "L", "XL"], label="taille", info="choisie la taille"
         ),
         ], outputs="image").launch(share=True, debug=True)
-