Update README.md

README.md

@@ -5,15 +5,15 @@ language:
 base_model: microsoft/phi-2
 pipeline_tag: text-generation
 ---
-https://arxiv.org/abs/1710.06071
-# Model Card for Model ID
 ![](ft_sections.png)
 
-This is a small language model designed for scientific research. It specializes in analyzing clinical trial abstracts and sorts sentences into four key sections: Background, Methods, Results, and Conclusion. 
+This is a small language model designed for scientific research application. It is fine tuned to analyzing randomized clinical trial abstracts and to classify sentences into four key sections: Background, Methods, Results, and Conclusion. 
 This makes it easier and faster for researchers to understand and organize important information from clinical studies.
 
 ## Model Details
 
+
+
 - **Developed by: Salvatore Saporito
 - **Language(s) (NLP):** English
 - **Finetuned from model:** https://huggingface.co/microsoft/phi-2
@@ -24,7 +24,7 @@ This makes it easier and faster for researchers to understand and organize impor
 
 ## Uses
 
-Automatic identification of sections in (clinical trial) abstracts.
+Automatic identification of sections in (randomized clinical trial) abstracts.
 
 ## How to Get Started with the Model
 
@@ -50,6 +50,17 @@ Usage:
     text = tokenizer.batch_decode(outputs,skip_special_tokens=True)[0]
     print(text)
 
+
+Example:
+Application on unseen data (
+
+ )
+
+        PROMPT: ['###Unstruct:\nKawasaki disease (KD) is a systemic vasculitis that causes abnormalities in the coronary arteries. Interleukin (IL)-41 is a novel immunoregulatory cytokine involved in the pathogenesis of some inflammatory and immune-related diseases. However, the role of IL-41 in KD is unclear. The purpose of this study was to detect the expression of IL-41 in the plasma of children with KD and its relationship with the disease.\nA total of 44 children with KD and 37 healthy controls (HC) were recruited for this study. Plasma concentrations of IL-41 were determined by ELISA. Correlations between plasma IL-41 levels and KD-related clinical parameters were analyzed by Pearson correlation and multivariate linear regression analysis. Receiver operating characteristic curve analysis was used to assess the clinical value of IL-41 in the diagnosis of KD.\nOur results showed that plasma IL-41 levels were significantly elevated in children with KD compared with HC. Correlation analysis demonstrated that IL-41 levels were positively correlated with D-dimer and N-terminal pro-B-type natriuretic peptide, and negatively correlated with IgM, mean corpuscular hemoglobin concentration, total protein, albumin and pre-albumin. Multivariable linear regression analysis revealed that IgM and mean corpuscular hemoglobin concentrations were associated with IL-41. Receiver operating characteristic curve analysis showed that the area under the curve of IL-41 was 0.7101, with IL-41 providing 88.64 % sensitivity and 54.05 % specificity.\nOur study indicated that plasma IL-41 levels in children with KD were significantly higher than those in HC, and may provide a potential diagnostic biomarker for KD.\n###Struct:\n
+
+        Answer Phi2_RCT1M-ft-heading:
+
+        BACKGROUND: Kawasaki disease (KD) is a systemic vasculitis that causes abnormalities in the coronary arteries. Interleukin (IL)-41 is a novel immunoregulatory cytokine involved in the pathogenesis of some inflammatory and immune-related diseases. However, the role of IL-41 in KD is unclear. The purpose of this study was to detect the expression of IL-41 in the plasma of children with KD and its relationship with the disease. METHODS: A total of 44 children with KD and 37 healthy controls (HC) were recruited for this study. Plasma concentrations of IL-41 were determined by ELISA. Correlations between plasma IL-41 levels and KD-related clinical parameters were analyzed by Pearson correlation and multivariate linear regression analysis. Receiver operating characteristic curve analysis was used to assess the clinical value of IL-41 in the diagnosis of KD. RESULTS: Our results showed that plasma IL-41 levels were significantly elevated in children with KD compared with HC. Correlation analysis demonstrated that IL-41 levels were positively correlated with D-dimer and N-terminal pro-B-type natriuretic peptide, and negatively correlated with IgM, mean corpuscular hemoglobin concentration, total protein, albumin and pre-albumin. Multivariable linear regression analysis revealed that IgM and mean corpuscular hemoglobin concentrations were associated with IL-41. Receiver operating characteristic curve analysis showed that the area under the curve of IL-41 was 0.7101, with IL-41 providing 88.64 % sensitivity and 54.05 % specificity. CONCLUSIONS: Our study indicated that plasma IL-41 levels in children with KD were significantly higher than those in HC, and may provide a potential diagnostic biomarker for KD.']
     
 ## Training Details
 
@@ -66,7 +77,7 @@ Generation of dedicated prompt for Causal_LM modelling.
 
 #### Training Hyperparameters
 
-bnb_config = BitsAndBytesConfig(load_in_4bit=True,
+    bnb_config = BitsAndBytesConfig(load_in_4bit=True,
                                 bnb_4bit_quant_type='nf4',
                                 bnb_4bit_compute_dtype=torch.bfloat16,
                                 bnb_4bit_use_double_quant=True)
@@ -83,13 +94,7 @@ bnb_config = BitsAndBytesConfig(load_in_4bit=True,
 
 #### Metrics
 
-
-
-### Results
-
-
-#### Summary
-
+Coming soon
 
 ## Technical Specifications [optional]
 
@@ -115,4 +120,9 @@ bnb_config = BitsAndBytesConfig(load_in_4bit=True,
 
 torch einops transformers bitsandbytes accelerate peft 
 
-## Model Card Contact
+## Model Card Contact
+
+
+## References
+
+https://arxiv.org/abs/1710.06071