Update README.md

Changed description. Added links guiding users to my final fine tuned model

README.md

@@ -1,3 +1,21 @@
+---
+base_model: microsoft/phi-2
+library_name: peft
+license: apache-2.0
+datasets:
+- neil-code/dialogsum-test
+language:
+- en
+metrics:
+- bleu
+pipeline_tag: question-answering
+tags:
+- QuestionAnswering
+- legal
+- finan
+- chem
+- biology
+---
 license: apache-2.0
 language:
 - en
@@ -6,8 +24,12 @@ metrics:
 base_model:
 - microsoft/phi-2
 pipeline_tag: question-answering
-library_name: fastai
 ---
+
+
+
+This repo containes the last checkpoint of my fine tuned model. Click this link to go the final model
+https://huggingface.co/JamieAi33/Phi-2_PEFT
 # Model Card for PEFT-Fine-Tuned Model
 
 This model card documents a PEFT-fine-tuned version of `microsoft/phi-2` for question-answering tasks. The PEFT fine-tuning improved the model's performance, as detailed in the evaluation section.
@@ -96,4 +118,4 @@ The model's performance was evaluated using the ROUGE metric. Below are the resu
 | **Optimizer**         | Paged AdamW (8-bit)    |
 | **Logging Steps**     | 25                     |
 | **Evaluation Steps**  | 25                     |
-| **Gradient Checkpointing** | Enabled          |
+| **Gradient Checkpointing** | Enabled          |