Update app.py

app.py

@@ -30,14 +30,28 @@ def generate_response(prompt, model=model, tokenizer=tokenizer, max_length=100,
         early_stopping=True
     )
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return response.strip()
+    
+    # Clean up the response to make it more natural
+    response = response.replace('User:', '').replace('Assistant:', '').strip()
+    return response
 
 # Define the Gradio interface
 def chatbot_interface(user_input):
     return generate_response(user_input)
 
-iface = gr.Interface(fn=chatbot_interface, inputs="text", outputs="text", title="Chatbot", description="Ask anything to the chatbot.")
+iface = gr.Interface(
+    fn=chatbot_interface,
+    inputs="text",
+    outputs="text",
+    title="Chatbot",
+    description="Ask anything to the chatbot.",
+    examples=[
+        ["What is the command to get a list of all files in /var/cache/apt/archives?"],
+        ["How do I change to root in Ubuntu?"],
+        ["What are the best practices for securing a web server?"]
+    ],
+    live=True
+)
 
 # Launch the Gradio interface
-iface.launch()
-
+iface.launch()