Update app.py

app.py

@@ -31,13 +31,13 @@ def useZephyr(prompt, promptdos):
     return outputs[0]["generated_text"]
 
 
-def generatePrompt(prompt):
+def generatePrompt(prompt, promptdos):
     batch = tokenizer(prompt, return_tensors="pt")
     generated_ids = model.generate(batch["input_ids"])
     output = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
     prompt = output[0]
 
-    result = useZephyr(prompt)
+    result = useZephyr(prompt, promptdos)
     
     return result 
 #
@@ -52,7 +52,7 @@ examples = [["photographer"], ["developer"], ["teacher"], [
     "human resources staff"], ["recipe for ham croquettes"]]
 description = ""
 
-PerfectGPT = gr.Interface(generatePrompt, inputs=input_prompt, outputs=output_component,
+PerfectGPT = gr.Interface(generatePrompt, inputs=[input_prompt,input_promptdos] outputs=output_component,
                           examples=examples, title="🗿 PerfectGPT v1 🗿", description=description)
 
 PerfectGPT.launch()