Update app.py

app.py

@@ -21,6 +21,7 @@ API_URL_DEV = "https://api-inference.huggingface.co/models/black-forest-labs/FLU
 API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
 
 chatter="K00B404/transcript_image_generator"
+max_tokens_enhance_bot=128
 # Initialize the API client for the chatbot
 chatbot_client = Client(chatter)
 
@@ -79,7 +80,7 @@ def enhance_prompt_v2(prompt, model="mistralai/Mistral-Nemo-Instruct-2407", styl
     result = client.predict(
     		system_prompt=system_prompt,
     		user_message=user_message,
-    		max_tokens=256,
+    		max_tokens=max_tokens_enhance_bot,
     		model_id=model,
     		api_name="/predict"
     )
@@ -101,7 +102,7 @@ def mistral_nemo_call(prompt, API_TOKEN, model="mistralai/Mistral-Nemo-Instruct-
         messages=[{"role": "system", "content": system_prompt},
                   {"role": "user", "content": prompt}
                  ],
-        max_tokens=500,
+        max_tokens=max_tokens_enhance_bot,
         stream=True,
     ):
         response += message.choices[0].delta.content
@@ -166,7 +167,7 @@ def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Ka
     You must respond only with the enhanced version of the users input prompt
     Remember, image generation models can be stimulated by refering to camera 'effect' in the prompt like :4k ,award winning, super details, 35mm lens, hd
     """
-        prompt = chat_with_persona(message=prompt, history=[], system_message=system_prompt, max_tokens=256, temperature=0.1, top_p=0.97)
+        prompt = chat_with_persona(message=prompt, history=[], system_message=system_prompt, max_tokens=max_tokens_enhance_bot, temperature=0.1, top_p=0.97)
         #prompt = enhance_prompt_v2(prompt, style=enhance_prompt_style)
         print(f'\033[1mGeneration {key} enhanced prompt:\033[0m {prompt}')
         
@@ -177,7 +178,7 @@ def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Ka
     You must respond only with the enhanced version of the users input prompt
     Remember, image generation models can be stimulated by refering to camera 'effect' in the prompt like :4k ,award winning, super details, 35mm lens, hd
     """
-        prompt = chat_with_persona(message=prompt, history=[], system_message=system_prompt, max_tokens=256, temperature=0.1, top_p=0.97)
+        prompt = chat_with_persona(message=prompt, history=[], system_message=system_prompt, max_tokens=max_tokens_enhance_bot, temperature=0.1, top_p=0.97)
         #prompt = mistral_nemo_call(prompt, API_TOKEN=API_TOKEN, style=nemo_enhance_prompt_style)
         print(f'\033[1mGeneration {key} Mistral-Nemo prompt:\033[0m {prompt}')
         
@@ -230,7 +231,7 @@ title_html="""
 css = """
 .gradio-container {
     background: url(https://huggingface.co/spaces/K00B404/FLUX.1-Dev-Serverless-darn-enhanced-prompt/resolve/main/edge.png);
-    background-size: 900px 880px;
+    background-size: 900px 1880px;
     background-repeat: no-repeat;
     background-position: center;
     background-attachment: fixed;