Update app.py

app.py

@@ -6,7 +6,7 @@ generator = pipeline('text-generation', model='gpt2')
 
 def generate_text(prompt, temperature, max_length, instruction):
     set_seed(42)
-    result = generator(prompt, max_length=max_length, num_return_sequences=1, temperature=temperature)
+    result = generator(prompt + "\nAI: ", max_length=max_length, num_return_sequences=1, temperature=temperature)
     return result[0]['generated_text']
 
 # Define Gradio interface
@@ -15,7 +15,7 @@ iface = gr.Interface(
     inputs=[gr.inputs.Textbox(lines=3, label="Your Message"),
             gr.inputs.Slider(minimum=0.1, maximum=1.0, default=0.5, label="Temperature"),
             gr.inputs.Slider(minimum=10, maximum=200, default=100, label="Max Length"),
-            gr.inputs.Textbox(default="This is a chat between the user and the AI.\nUser: hi\nAI: Hello, what do you need assistance with?\nUser: ", lines=2, label="AI Instruction")],
+            gr.inputs.Textbox(default="This is a chat between the user and a truthful and intelligent AI. AI must answer very truthfully and accurately. The chat begins now.\nUser: hi\nAI: Hello, what do you need assistance with?\nUser: ", lines=2, label="AI Instruction")],
     outputs=gr.outputs.Textbox(label="AI Response"),
     layout="vertical",
     title="Chat with GPT-2",