Update app.py

app.py

@@ -14,19 +14,20 @@ pipe = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-alpha",
                 torch_dtype=torch.bfloat16, device_map="auto")
 
 
-def useZephyr(prompt):
+def useZephyr(prompt, promptdos):
     messages = [
         {
             "role": "system",
             "content": "you are a chatbot who always responds politely and in the shortest possible way",
         },
-        {"role": "user", "content": "{prompt}"},
+        {"role": "user", "content": prompt},
+        {"role": "user", "content": promptdos},
     ]
     # https://huggingface.co/docs/transformers/main/en/chat_templating
-    prompt = pipe.tokenizer.apply_chat_template(
+    new_prompt = pipe.tokenizer.apply_chat_template(
         messages, tokenize=False, add_generation_prompt=True)
 
-    outputs = pipe(prompt)
+    outputs = pipe(new_prompt)
     return outputs[0]["generated_text"]
 
 
@@ -44,8 +45,9 @@ def generatePrompt(prompt):
 # Interface
 
 
-input_prompt = gr.Textbox(label="Prompt", value="photographer")
-output_component = gr.Textbox(label="Output")
+input_prompt = gr.Textbox(label="Actua como: ", value="Chef")
+input_promptdos = gr.Textbox(label="Prompt: ", value="Recipe for ham croquettes")
+output_component = gr.Textbox(label="Output: ")
 examples = [["photographer"], ["developer"], ["teacher"], [
     "human resources staff"], ["recipe for ham croquettes"]]
 description = ""