Actualizada la función del chatbot

app.py

@@ -18,7 +18,7 @@ def useZephyr(prompt):
     messages = [
         {
             "role": "system",
-            "content": "You are a friendly chatbot who always responds in the style of a pirate.",
+            "content": "you are a chatbot who always responds politely and in the shortest possible way",
         },
         {"role": "user", "content": prompt},
     ]
@@ -26,7 +26,6 @@ def useZephyr(prompt):
     prompt = pipe.tokenizer.apply_chat_template(
         messages, tokenize=False, add_generation_prompt=True)
 
-    # outputs = pipe(prompt, max_new_tokens=256, do_sample=True,temperature=0.7, top_k=50, top_p=0.95)
     outputs = pipe(prompt)
     return outputs[0]["generated_text"]
 
@@ -37,7 +36,9 @@ def generatePrompt(prompt):
     output = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
     prompt = output[0]
 
-    return useZephyr(prompt)
+    result = useZephyr(prompt)
+    
+    return result 
 #
 
 # Interface