Update app.py

app.py

@@ -2,11 +2,15 @@ import torch
 from transformers import BartForConditionalGeneration, BartTokenizer
 import gradio as gr
 
+ 
+
 # Load the fine-tuned model and tokenizer
 model_path = "Asna-DifiNative/AIBuddy"  # Path to the pretrained fine-tuned model
 model = BartForConditionalGeneration.from_pretrained(model_path)
 tokenizer = BartTokenizer.from_pretrained(model_path)
 
+ 
+
 # Translate function using the fine-tuned model
 def translate_instruction(context, input_text):
     full_input = context + " " + input_text
@@ -16,15 +20,20 @@ def translate_instruction(context, input_text):
     translated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return translated_text
 
+
+ 
+
 # Example instructions
 example_instructions = [
-    ["Pod xyz is called as tiger.", "Show me a list of pods in the current namespace."],
-    ["Nickname of deployment development is cap", "Display the pods running in the cap."],
-    ["Production is also known as", "List all pods in the 'production' environment."],
-    ["Kubernetes", "Show the pods running in the 'testing' namespace?"],
-    ["Docker", "What command should I use to get the list of containers?"]
+    "Show me a list of pods in the current namespace.",
+    "Display the pods running in the 'development' namespace.",
+    "List all pods in the 'production' environment.",
+    "How can I see the pods in the 'testing' namespace?",
+    "What command should I use to get the list of containers?"
 ]
 
+ 
+
 # Gradio Interface
 iface = gr.Interface(
     fn=translate_instruction,
@@ -34,18 +43,18 @@ iface = gr.Interface(
     ],
     outputs=gr.outputs.Textbox(label="Generated CLI Command"),
     title="CLI Command Generator",
-    description=f"<p>This tool generates CLI commands from human instructions and context.</p>"
-                f"<p>Provide a context and human instruction, and click 'Generate' to get the CLI command.</p>"
-                f"<h3>Example Instructions:</h3>"
-                f"<table style='text-align: left;'>"
-                f"<tr><td>Pod xyz is called as tiger.</td><td>Show me a list of pods in the current namespace.</td></tr>"
-                f"<tr><td>Nickname of deployment development is cap</td><td>Display the pods running in the cap.</td></tr>"
-                f"<tr><td>Production is also known as</td><td>List all pods in the 'production' environment.</td></tr>"
-                f"<tr><td>Kubernetes</td><td>Show the pods running in the 'testing' namespace?</td></tr>"
-                f"<tr><td>Docker</td><td>What command should I use to get the list of containers?</td></tr>"
-                f"</table>",
-    examples=example_instructions
+    description="<p>This tool generates CLI commands from human instructions and context.</p>"
+                "<p>Provide a context and human instruction, and click 'Generate' to get the CLI command.</p>",
+    examples=[
+        ["Pod xyz is called as tiger.", "Show me a list of pods in the current namespace."],
+        ["Nickname of deployment development is cap", "Display the pods running in the cap."],
+        ["Production is also known as", "List all pods in the 'production' environment."],
+        ["Kubernetes", "Show the pods running in the 'testing' namespace?"],
+        ["Docker", "What command should I use to get the list of containers?"]
+    ]
 )
 
+ 
+
 # Launch the Gradio app using Ngrok
-iface.launch()
+iface.launch()