MaxMilan1
commited on
Commit
·
36be636
1
Parent(s):
008cf2e
more changees
Browse files- app.py +0 -1
- 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 =
|
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 |
|