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 = "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
@@ -15,6 +19,7 @@ def translate_instruction(context, input_text):
         outputs = model.generate(input_ids, max_length=50, num_beams=4, early_stopping=True)
     translated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return translated_text
+ 
 
 # Gradio Interface
 iface = gr.Interface(
@@ -27,17 +32,16 @@ iface = gr.Interface(
     title="CLI Command Generator",
     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="""<div style='text-align: left'>
-        <p>Examples:</p>
-        <ul>
-            <li>Pod xyz is called as tiger. -> Show me a list of pods in the current namespace.</li>
-            <li>Nickname of deployment development is cap -> Display the pods running in the cap.</li>
-            <li>Production is also known as -> List all pods in the 'production' environment.</li>
-            <li>Kubernetes -> Show the pods running in the 'testing' namespace?</li>
-            <li>Docker -> What command should I use to get the list of containers?</li>
-        </ul>
-        </div>"""
+    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()