artificialguybr
commited on
Commit
•
3e5f8f0
1
Parent(s):
68c1454
Update app.py
Browse files
app.py
CHANGED
@@ -17,8 +17,9 @@ def generate_image(prompt):
|
|
17 |
headers = {
|
18 |
#"Authorization": f"Bearer {token}"
|
19 |
}
|
|
|
20 |
payload = {
|
21 |
-
"inputs":
|
22 |
"parameters": {
|
23 |
"negative_prompt": "bad art, ugly, watermark, deformed",
|
24 |
"num_inference_steps": 30,
|
@@ -66,4 +67,4 @@ iface = gr.Interface(
|
|
66 |
)
|
67 |
|
68 |
# Iniciando a interface
|
69 |
-
iface.launch()
|
|
|
17 |
headers = {
|
18 |
#"Authorization": f"Bearer {token}"
|
19 |
}
|
20 |
+
full_prompt = f"{prompt} {trigger_word}" # Combina o prompt do usuário com a trigger word
|
21 |
payload = {
|
22 |
+
"inputs": full_prompt,
|
23 |
"parameters": {
|
24 |
"negative_prompt": "bad art, ugly, watermark, deformed",
|
25 |
"num_inference_steps": 30,
|
|
|
67 |
)
|
68 |
|
69 |
# Iniciando a interface
|
70 |
+
iface.launch()
|