Update app.py

app.py

@@ -4,7 +4,9 @@ from transformers import pipeline
 
 def generate_text(prompt, temperature):
     generator = pipeline('text-generation', model='titleos/Seahorse-350m')
-    return generator(prompt, max_length=100, temperature=temperature)[0]['generated_text']
+    generated_text = generator(prompt, max_length=100, temperature=temperature)[0]['generated_text']
+    response = generated_text.split("### Response")[-1]
+    return response
 
 interface = gr.Interface(
     fn=generate_text,