more changes

app.py

@@ -4,8 +4,13 @@ import torch
 from diffusers import DiffusionPipeline
 import rembg
 
-model_id = "stabilityai/stable-diffusion-2-1"
-pipe = DiffusionPipeline.from_pretrained(model_id)
+model_id = "stabilityai/stable-diffusion-xl-base-1.0"
+pipe = DiffusionPipeline.from_pretrained(model_id,
+                                         torch_dtype=torch.float16,
+                                        use_safetensors=True,)
+
+pipe.load_lora_weights("CiroN2022/shoes", weight_name="shoes.safetensors", adapter_name="shoes")
+pipe.set_adapters("shoes")
 
 pipe.to("cuda")
 
@@ -14,20 +19,21 @@ pipe.to("cuda")
 def generate_image(prompt):
     
     image = pipe(prompt).images
-    image = rembg.remove(image)
-    return image
+    image2 = rembg.remove(image)
+    return image, image2
 
 _TITLE = "Shoe Generator"
 with gr.Blocks(_TITLE) as ShoeGen:
     with gr.Row():
         with gr.Column():
             prompt = gr.Textbox(label="Enter a prompt")
+            neg_prompt = gr.Textbox(label="Enter a negative prompt", value="low quality, watermark, ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, extra limbs, body out of frame, blurry, bad anatomy, blurred, watermark, grainy, signature, cut off, draft, closed eyes, text, logo")
             button_gen = gr.Button("Generate Image")
         with gr.Column():
-        #  show images
-            image = gr.Image(label="Generated Image", show_download_button=True)    
+            image = gr.Image(label="Generated Image", show_download_button=True)   
+            image2 = gr.Image(label="Generated Image without background", show_download_button=True) 
     
 
-    button_gen.click(generate_image, inputs=[prompt], outputs=image)
+    button_gen.click(generate_image, inputs=[prompt], outputs=[image, image2])
 
 ShoeGen.launch()

requirements.txt

@@ -3,4 +3,5 @@ gradio
 diffusers==0.26.3
 transformers==4.38.1
 accelerate==0.27.2
-xformers
+xformers
+rembg