Update README.md

README.md

@@ -1,10 +1,15 @@
 ---
 license: other
-tags:
-- generated_from_trainer
 model-index:
 - name: Doctor-OPT-350M
   results: []
+widget:
+- text: |-
+    Patient: I've been experiencing a persistent cough for the past week. What could be causing it?
+
+    Doctor: <start>
+datasets:
+- sid6i7/patient-doctor
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -12,39 +17,59 @@ should probably proofread and complete it, then remove this comment. -->
 
 # Doctor-OPT-350M
 
-This model is a fine-tuned version of [facebook/opt-350m](https://huggingface.co/facebook/opt-350m) on an unknown dataset.
+This model is a fine-tuned version of [facebook/opt-350m](https://huggingface.co/facebook/opt-350m) on the [sid6i7/patient-doctor](https://huggingface.co/datasets/sid6i7/patient-doctor).
 
 ## Model description
 
-More information needed
+This is an LLM trained on patient-doctor dialogues, mostly comprised of doctors giving medical advice to patients.
 
 ## Intended uses & limitations
 
-More information needed
+This model is intended for research on the capabilities of LLMs. Any output from this LLM is not to be treated as legitimate medical advice.
+
+The model works best when given input in the format of the dataset it was trained on. Here is an example of the format:
+
+```
+Patient: Please can you help me with thoughts about its. Where my gut gives out spasms and. Uncontrol movements like pulsating 24,7. It seems. Worst when stressed I would like something that regards relaxing the bowel to get some more pleasures in life. Many thanks. XXXX
+
+Doctor: <start> Hello, Continuous abdominal spasms can be due to different reasons. Is there any intense pain present? Are you on any medication?It might be related to food habits. Try consuming food that is rich in fibers.
+```
+
+
+## Sample Use
+
+```python
+from transformers import pipeline
+
+generator = pipeline('text-generation', model='linkanjarad/Doctor-OPT-350M')
+
+# Format the prompt to match the dataset formatting
+prompt = '''Patient: Hi, I'm 30 years old. Recently, I've been having abdominal pain and discomfort after meals. It feels like a burning sensation. What could be the possible causes?
+
+Doctor: <start>'''
+
+output = generator(prompt, do_sample=True, min_length=10, max_length=50, temperature=0.5)
+print(output[0]['generated_text'])
+```
 
-## Training and evaluation data
+## Sample Outputs
 
-More information needed
+```
+Patient: "I'm a 40-year-old male with a family history of heart disease. Lately, I've been experiencing chest discomfort and shortness of breath during physical activity. Are these symptoms indicative of a heart problem, and what tests should I undergo to evaluate my heart health?
 
-## Training procedure
+Doctor: <start> Hello,The symptoms you have mentioned are suggestive of cardiac problems. These symptoms are also possible due to high blood pressure and cholesterol. I would recommend performing physical activity such as walking, rollerblading, swimming, etc. to increase your activity levels.I would also recommend doing an ultrasound scan of the heart to determine if there is a risk of a heart problem. I would also recommend doing a blood test to
+```
 
-### Training hyperparameters
+```
+Patient: I'm feeling this pain in my chest when I see this particular person. My breath feels shortened and I feel nervous, and I get a tingly sensation. My eyes seem to follow the person. What could this be?
 
-The following hyperparameters were used during training:
-- learning_rate: 2e-05
-- train_batch_size: 1
-- eval_batch_size: 1
-- seed: 42
-- gradient_accumulation_steps: 256
-- total_train_batch_size: 256
-- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
-- lr_scheduler_type: linear
-- lr_scheduler_warmup_ratio: 0.15
-- num_epochs: 1
+Doctor: <start> Hi,
+ Thanks for writing in.
+ The symptoms you have described suggest to be related to some sort of pulmonary infections. I would advise you to undergo a check up by a physician of the chest and abdomen. In addition, you should go for a chest X-ray and a chest and abdomen MRI.
+```
 
-### Framework versions
+```
+Patient: Hello, I am a 20 year old female and have been getting stabbing pains on the lower right side of my abdomen everyday for about 4 weeks now. I know it cant be my appendix for it was removed in 2010. I have noticed that above the scar of my appendix removal the area seems risen, and beneath the scar the area is flat. Should i be worried and seek further help?
 
-- Transformers 4.27.4
-- Pytorch 1.13.0
-- Datasets 2.1.0
-- Tokenizers 0.13.2
+Doctor: <start> Hi dear,Thanks for writing in.The pain in the upper right side of the abdomen may be because of some sort of appendicitis. Some times it can be due to a condition called anemia. If there is anemia and the pain is not related to anemia, then chances of appendicitis are low. I would suggest you to consult an emic specialist and get a urine and blood test done. Anemia can be treated in a few ways. For example, erythrocyte sedimentation rate (ESR), blood protein level, and a few more tests like CT scan of the abdomen and kidney are important but the actual diagnosis is not always confirmed. I hope my answer will help you.Wish you all the best.
+```