Update app.py

app.py

@@ -14,7 +14,9 @@ pipe = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-alpha",
                 torch_dtype=torch.bfloat16, device_map="auto")
 
 
-def generatePrompt(prompt, promptdos):
+def generatePrompt(inputs):
+    inputs[0] = prompt
+    inputs[1] = promptdos
     batch = tokenizer(prompt, return_tensors="pt")
     generated_ids = model.generate(batch["input_ids"])
     output = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)