Update app.py
Browse files
app.py
CHANGED
@@ -8,27 +8,25 @@ import time
|
|
8 |
|
9 |
# Defining the repository information and the trigger word
|
10 |
repo = "artificialguybr/TshirtDesignRedmond-V2"
|
11 |
-
trigger_word = "T shirt design, TshirtDesignAF, "
|
12 |
-
|
13 |
# Function to generate image based on the prompt
|
14 |
def infer(color_prompt, dress_type_prompt, design_prompt, text):
|
15 |
# Build the full prompt
|
16 |
prompt = (
|
17 |
-
f"a single {color_prompt} colored {dress_type_prompt} with {design_prompt}
|
18 |
"hangs effortlessly on a plain wall, its simplicity transformed by bold,"
|
19 |
)
|
20 |
|
21 |
# Conditional parts
|
22 |
if text:
|
23 |
prompt += (
|
24 |
-
f"
|
25 |
"The contrast between the text and the calm background creates a striking visual"
|
26 |
)
|
27 |
|
28 |
# Add the hidden shadows part
|
29 |
prompt += f"soft light casts dynamic shadows, adding depth and emphasizing the crisp lines of the design, evoking a sense of modern sophistication."
|
30 |
|
31 |
-
full_prompt = f"{prompt}
|
32 |
|
33 |
print("Generating image with prompt:", full_prompt)
|
34 |
api_url = f"https://api-inference.huggingface.co/models/{repo}"
|
|
|
8 |
|
9 |
# Defining the repository information and the trigger word
|
10 |
repo = "artificialguybr/TshirtDesignRedmond-V2"
|
|
|
|
|
11 |
# Function to generate image based on the prompt
|
12 |
def infer(color_prompt, dress_type_prompt, design_prompt, text):
|
13 |
# Build the full prompt
|
14 |
prompt = (
|
15 |
+
f"a single {color_prompt} colored {dress_type_prompt} with {design_prompt} designed on the {color_prompt} colored {dress_type_prompt}"
|
16 |
"hangs effortlessly on a plain wall, its simplicity transformed by bold,"
|
17 |
)
|
18 |
|
19 |
# Conditional parts
|
20 |
if text:
|
21 |
prompt += (
|
22 |
+
f" and {text} written on the {dress_type_prompt}"
|
23 |
"The contrast between the text and the calm background creates a striking visual"
|
24 |
)
|
25 |
|
26 |
# Add the hidden shadows part
|
27 |
prompt += f"soft light casts dynamic shadows, adding depth and emphasizing the crisp lines of the design, evoking a sense of modern sophistication."
|
28 |
|
29 |
+
full_prompt = f"{prompt}"
|
30 |
|
31 |
print("Generating image with prompt:", full_prompt)
|
32 |
api_url = f"https://api-inference.huggingface.co/models/{repo}"
|