Upload app.py

app.py

@@ -35,6 +35,16 @@ def generate(prompt, max_length, temperature, num_samples):
 
 with gr.Blocks() as app:
     gr.Markdown("## ERA Session27 - Phi2 Model Finetuning with QLoRA on OpenAssistant Conversations Dataset (OASST1)")
+    gr.Markdown(
+        """This is an implementation of [Phi2](https://huggingface.co/microsoft/phi-2) model finetuning using QLoRA stratergy on [OpenAssistant Conversations Dataset (OASST1)](https://huggingface.co/datasets/OpenAssistant/oasst1)
+        
+        Please find the source code and training details [here](https://github.com/RaviNaik/ERA-SESSION27).
+        
+        Dataset used to finetune: [OpenAssistant Conversations Dataset (OASST1)](https://huggingface.co/datasets/OpenAssistant/oasst1)
+        ChatML modified OSST Dataset: [RaviNaik/oasst1-chatml](https://huggingface.co/datasets/RaviNaik/oasst1-chatml)
+        Finetuned Model: [RaviNaik/Phi2-Osst](https://huggingface.co/RaviNaik/Phi2-Osst)
+        """
+    )
     with gr.Row():
         with gr.Column():
             prompt_box = gr.Textbox(label="Initial Prompt", interactive=True)