Update app.py

app.py

@@ -4,16 +4,10 @@ import torch
 from inference import run_inference
 
 
-description_zero_shot_training = """ ### Zero Shot Training
-                
-                1. Choose a Dataset MNIST/CIFAR10
-                
-                2. Output will be class accuracy
-                                
-              """
+description_zero_shot_training = """ ### Chat with Phi-2 """
 
 # Description
-title = "<center><strong><font size='8'>📎 THE CLIP PLAYGROUND 📎</font></strong></center>"
+title = "<center><strong><font size='8'>📎 Phi-2 Chat (Trained on OpenAssistant/oasst1 Dataset with QLoRA) 📎</font></strong></center>"
 
 text_input = gr.Text(label="Enter text")
 text_input2 = gr.Text(label="Generated Response")
@@ -38,7 +32,7 @@ with gr.Blocks(css=css, title='Play with CLIP') as demo:
         # Submit & Clear
         with gr.Row():
             with gr.Column():
-                run_chat_with_phi2_button = gr.Button("chat_with_phi2", variant='primary')
+                run_chat_with_phi2_button = gr.Button("chat with phi2", variant='primary')
                 clear_btn_text_to_image = gr.Button("Clear", variant="secondary")
                 gr.Markdown(description_zero_shot_training)
                 gr.Examples(examples = ["What is Large Language models ?", "Can you write a short introduction about the relevance of the term monopsony in economics? Please use examples related to potential monopsonies in the labour market and cite relevant research.", "I want to start doing astrophotography as a hobby, any suggestions what could i do?"],