Update app.py

app.py

@@ -14,9 +14,9 @@ pipe = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-alpha",
                 torch_dtype=torch.bfloat16, device_map="auto")
 
 
-def generatePrompt(inputs):
-    prompt = inputs[0] 
-    promptdos = inputs[1]
+def generatePrompt(input, inputdos):
+    prompt = input
+    promptdos = inputdos
     batch = tokenizer(prompt, return_tensors="pt")
     generated_ids = model.generate(batch["input_ids"])
     output = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
@@ -27,8 +27,12 @@ def generatePrompt(inputs):
             "role": "system",
             "content": "you are a chatbot who always responds politely and in the shortest possible way",
         },
-        {"role": "user", "content": str(new_prompt)},
-        {"role": "user", "content": str(promptdos)},
+        {
+            "role": "user", "content": str(new_prompt)
+        },
+        {
+            "role": "user", "content": str(promptdos)
+        },
     ]
     # https://huggingface.co/docs/transformers/main/en/chat_templating
     final_prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)