Update app.py

app.py

@@ -16,7 +16,7 @@ def format_prompt(new_message, history):
     prompt += f"[INST] {new_message} [/INST]"
     return prompt
 
-def generate(prompt, history, temperature=0.25, max_new_tokens=512, top_p=0.95, repetition_penalty=1.0):
+def generate(prompt, history, temperature=0.25, max_new_tokens=1024, top_p=0.95, repetition_penalty=1.0):
     # Konfiguration der Parameter
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -46,6 +46,6 @@ samir_chatbot = gr.Chatbot(bubble_full_width=True, show_label=False, show_copy_b
 
 # Minimalistisches Theme und Konfiguration der Gradio-Demo
 theme = 'syddharth/gray-minimal'
-demo = gr.ChatInterface(fn=generate, chatbot=samir_chatbot, title="Mistral Chat Prompt Generator ", theme=theme)
+demo = gr.ChatInterface(fn=generate, chatbot=samir_chatbot, title="WP-Plugin Generator", theme=theme)
 
 demo.queue().launch(show_api=False)