K00B404 commited on
Commit
1ea0b01
1 Parent(s): e725f65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -77,7 +77,7 @@ def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Ka
77
  prompt = enhance_prompt(prompt)
78
  print(f'\033[1mGeneration {key} enhanced prompt:\033[0m {prompt}')
79
  if use_mistral_nemo:
80
- prompt = mistral_nemo_call(prompt,API_TOKEN=API_TOKEN, model='meta-llama/Llama-3.2-1B',style="cartoon")
81
  print(f'\033[1mGeneration {key} Mistral-Nemo prompt:\033[0m {prompt}')
82
 
83
  final_prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
 
77
  prompt = enhance_prompt(prompt)
78
  print(f'\033[1mGeneration {key} enhanced prompt:\033[0m {prompt}')
79
  if use_mistral_nemo:
80
+ prompt = mistral_nemo_call(prompt,API_TOKEN=API_TOKEN,style="cartoon")
81
  print(f'\033[1mGeneration {key} Mistral-Nemo prompt:\033[0m {prompt}')
82
 
83
  final_prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."