yield responses

app.py

@@ -48,6 +48,8 @@ def generate_instruction_response():
         len(extract_input) :
     ].split("\n")[0]
 
+    yield "## Generated instructions: \n" + sanitized_instruction
+
     response_template = f"""<|begin_of_text|><|start_header_id|>user<|end_header_id|>\n\n{sanitized_instruction}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n"""
 
     response = pipeline(
@@ -59,10 +61,9 @@ def generate_instruction_response():
         top_p=1,
     )
 
-    user_message = sanitized_instruction
     assistant_response = response[0]["generated_text"][len(response_template) :]
 
-    return user_message, assistant_response
+    yield "## Generated response: \n" + assistant_response
 
 
 title = "Magpie demo"
@@ -77,10 +78,7 @@ You can learn more about the approach [in the paper](https://huggingface.co/pape
 iface = gr.Interface(
     fn=generate_instruction_response,
     inputs=[],
-    outputs=[
-        gr.Markdown(label="Generated User Instruction"),
-        gr.Markdown(label="Generated Model Response"),
-    ],
+    outputs=[gr.Markdown("Generated data")],
     title=title,
     description=description,
     submit_btn="Generate Instructions Response Pair",