Update app.py

app.py

@@ -44,7 +44,7 @@ def generate_response(message, max_tokens, temperature, top_p):
     global stop_signal
     stop_signal = False
 
-    prompt = f"Contexto: \nPensamento: \nEntrada: {message}\nResposta:"
+    prompt = f"Contexto: \nEntrada: {message}\nResposta:"
     inputs = tokenizer(prompt, return_tensors="pt").to(device)
 
     streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
@@ -75,14 +75,16 @@ def generate_response(message, max_tokens, temperature, top_p):
 
 # Interface Gradio
 with gr.Blocks(css=css, theme="NoCrypt/miku") as app:
-    chatbot = gr.Chatbot(label="λ", elem_id="chatbot")
+    chatbot = gr.Chatbot(label="WNL468M", elem_id="chatbot")
     msg = gr.Textbox(label="Mensagem", placeholder="Digite aqui...", lines=2)
     send_btn = gr.Button("Enviar")
     stop_btn = gr.Button("Parar")
 
-    max_tokens = gr.Slider(64, 512, value=128, step=1, label="Max Tokens")
-    temperature = gr.Slider(0.1, 1.5, value=0.7, step=0.1, label="Temperature")
-    top_p = gr.Slider(0.1, 1.0, value=0.95, step=0.05, label="Top-p")
+
+    gr.Row():
+        max_tokens = gr.Slider(64, 512, value=128, step=1, label="Max Tokens")
+        temperature = gr.Slider(0.1, 1.5, value=0.7, step=0.1, label="Temperature")
+        top_p = gr.Slider(0.1, 1.0, value=0.9, step=0.05, label="Top-p")
 
     state = gr.State([])  # histórico apenas visual