MaxMilan1 commited on
Commit
36be636
·
1 Parent(s): 008cf2e

more changees

Browse files
Files changed (2) hide show
  1. app.py +0 -1
  2. util/text_img.py +2 -2
app.py CHANGED
@@ -25,7 +25,6 @@ theme = gr.themes.Soft(
25
 
26
  with gr.Blocks(theme=theme) as GenDemo:
27
  with gr.Tab("Image to Image Generator"):
28
- button_choice = gr.Radio(label="Choose a model", choices=["Text to Image", "Image to Image"], value="Text to Image")
29
  with gr.Row(variant="panel"):
30
  with gr.Column():
31
  prompt = gr.Textbox(label="Enter a discription of a shoe")
 
25
 
26
  with gr.Blocks(theme=theme) as GenDemo:
27
  with gr.Tab("Image to Image Generator"):
 
28
  with gr.Row(variant="panel"):
29
  with gr.Column():
30
  prompt = gr.Textbox(label="Enter a discription of a shoe")
util/text_img.py CHANGED
@@ -59,9 +59,9 @@ def generate_txttoimg(prompt, control_image, controlnet):
59
 
60
  @spaces.GPU
61
  def generate_imgtoimg(prompt, image, strength=0.5):
62
- prompt += "no background, side view, minimalist shot, single shoe, no legs, product photo"
63
 
64
- image = pipeline(prompt, image=image, strength=strength).images[0]
65
 
66
  image2 = rembg.remove(image)
67
 
 
59
 
60
  @spaces.GPU
61
  def generate_imgtoimg(prompt, image, strength=0.5):
62
+ prompt += ", no background, side view, minimalist shot, single shoe, no legs, product photo"
63
 
64
+ image = imagepipe(prompt, image=image, strength=strength).images[0]
65
 
66
  image2 = rembg.remove(image)
67