artificialguybr commited on
Commit
3e5f8f0
1 Parent(s): 68c1454

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -17,8 +17,9 @@ def generate_image(prompt):
17
  headers = {
18
  #"Authorization": f"Bearer {token}"
19
  }
 
20
  payload = {
21
- "inputs": f"{prompt} {trigger_word}",
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()