salomonsky commited on
Commit
d801794
1 Parent(s): 06b8bc2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -63,9 +63,9 @@ async def gen(prompt, basemodel, width, height, scales, steps, seed, upscale_fac
63
 
64
  async def improve_prompt(prompt):
65
  try:
66
- instruction = "Improve and translate this prompt into English, adding detailed descriptions of style, cinematography, cameras, atmosphere, and lighting for the best quality, up to 200 words."
67
  formatted_prompt = f"{instruction}: {prompt}"
68
- response = llm_client.text_generation(formatted_prompt, max_new_tokens=300) # Allowing more tokens for detailed description
69
  improved_text = response['generated_text'].strip() if 'generated_text' in response else response.strip()
70
 
71
  return improved_text
 
63
 
64
  async def improve_prompt(prompt):
65
  try:
66
+ instruction = "Improve and translate this prompt into English, adding detailed descriptions of character, style, cinematography, cameras, atmosphere, and lighting for the best quality, up max to 300 words."
67
  formatted_prompt = f"{instruction}: {prompt}"
68
+ response = llm_client.text_generation(formatted_prompt, max_new_tokens=300)
69
  improved_text = response['generated_text'].strip() if 'generated_text' in response else response.strip()
70
 
71
  return improved_text