changeeeeeeeeeeeeeeeeeeeeeeees

app.py

@@ -28,7 +28,7 @@ with gr.Blocks(theme=theme) as GenDemo:
         with gr.Row(variant="panel"):
             with gr.Column():
                 prompt = gr.Textbox(label="Enter a discription of a shoe")
-                image = gr.Image(label="Enter an image of a shoe, that you want to use as a reference", type='pil')
+                image = gr.Image(label="Enter an image of a shoe, that you want to use as a reference")
                 strength = gr.Slider(label="Strength", minimum=0.1, maximum=1.0, value=0.5, step=0.1)
                 gr.Examples(
                        examples=[

util/text_img.py

@@ -14,10 +14,6 @@ import gradio as gr
 def check_prompt(prompt):
     if prompt is None:
         raise gr.Error("Please enter a prompt!")
-    
-imagepipe = AutoPipelineForImage2Image.from_pretrained(
-    "stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
-)
 
 controlNet_normal = ControlNetModel.from_pretrained(
         "fusing/stable-diffusion-v1-5-controlnet-normal", 
@@ -58,10 +54,14 @@ def generate_txttoimg(prompt, control_image, controlnet):
     return image2
 
 @spaces.GPU
-def generate_imgtoimg(prompt, image, strength=0.5):
+def generate_imgtoimg(prompt, init_image, strength=0.5):
     prompt += ", no background, side view, minimalist shot, single shoe, no legs, product photo"
     
-    image = imagepipe(prompt, image=image, strength=strength).images[0]
+    imagepipe = AutoPipelineForImage2Image.from_pretrained(
+    "stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
+    )
+
+    image = imagepipe(prompt, image=init_image, strength=strength).images[0]
     
     image2 = rembg.remove(image)