Update README.md

README.md

@@ -7,13 +7,13 @@ language:
 ---
 ## Overview
 
-This model, elucidator8918/apigen-prototype-0.1, is tailored for API generation, based on the Mistral-7B-Instruct-v0.1-sharded architecture fine-tuned on the LLAMA-2 Instruct 121k Code dataset.
+This model, elucidator8918/clinical-ehr-prototype-0.1, is tailored for clinical documentation, based on the Mistral-7B-Instruct-v0.1-sharded architecture fine-tuned on the Asclepius-Synthetic-Clinical-Notes dataset.
 
 ## Key Information
 
 - **Model Name**: Mistral-7B-Instruct-v0.1-sharded
 - **Fine-tuned Model Name**: elucidator8918/apigen-prototype-0.1
-- **Dataset**: emre/llama-2-instruct-121k-code
+- **Dataset**: starmpcc/Asclepius-Synthetic-Clinical-Notes
 - **Language**: English (en)
 
 ## Model Details
@@ -51,48 +51,40 @@ This model, elucidator8918/apigen-prototype-0.1, is tailored for API generation,
 ```python
 from transformers import pipeline
 
-api_gen_pipeline = pipeline("text-generation", model="elucidator8918/apigen-prototype-0.1")
+pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer)
 
 # Run text generation pipeline with our next model
-prompt = "Write code to do a POST request in FastAPI framework to find the multiplication of two matrices using NumPy"
-pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=500)
-result = pipe(f"[INST] {prompt} [/INST]")
-print(result[0]['generated_text'])
+prompt = """
+You are an intelligent clinical languge model.
+Below is a snippet of patient's electronic health record note and a following instruction with question from healthcare professional.
+Write a response that appropriately completes the instruction.
+The response should provide the accurate answer to the instruction, while being concise.
+
+### Instruction:
+Abbreviation Expansion
+
+### Patient's Electronic Health Record Note:
+Hospital Course: 
+
+This 66-year-old male patient was admitted due to an ischemic left-hemispheric stroke in addition to a dry cough and fever. The patient tested positive for SARS-CoV-2 and experienced severe ARDS, resulting in intubation and ICU admission. The patient underwent veno-venous extracorporeal membrane oxygenation and physical therapy was initiated to focus on perception training, movement exercises, airway-clearing techniques, dysphagia therapy, and mobilization. Despite a trial of sedation cessation, the patient remained somnolent and unable to communicate or follow commands. A side-edge positioning was initiated in combination with intensive exercise training including trunk and head control. Muscle tone and strength remained severely reduced, particularly on his hemiplegic side, and a second SOEB trial failed. Occupational therapy was involved to support functional initiation of upper limb movements and to integrate perception-training into activities of daily living. Currently, the patient remains functionally dependent, tolerates spontaneous breathing trials, and is alert during therapy, although he cannot communicate. He is considered stable and functionally dependent (CPAx 6/50).
+
+### Question:
+What are the abbreviated terms in the given discharge summary that require expansion?
+"""
+result = pipe(f"[INST] {prompt} [/INST]",max_length=584)[0]['generated_text']
+start_index = result.find("[/INST]") + len("[/INST]")
+end_index = result.find("'", start_index)
+response = result[start_index:end_index]
+print(response)
 ```
 
-- **Output API Generation:**
+- **Output Generation:**
 ```
-[INST] Write code to do a POST request in fastapi framework to find the multiplication of two matrices using numpy [/INST]
-Below is an example of how to make a POST request in FastAPI to find the multiplication of two matrices using numpy:
-```
-```python
-from fastapi import FastAPI, HTTPException
-import numpy as np
-
-app = FastAPI()
-
-@app.post("/matrix_multiplication")
-async def matrix_multiplication(matrix1: np.ndarray, matrix2: np.ndarray):
-    if matrix1.shape[1]!= matrix2.shape[0]:
-        raise HTTPException(status_code=400, detail="The number of columns in matrix1 must be equal to the number of rows in matrix2")
-    result = np.matmul(matrix1, matrix2)
-    return {"result": result}
-```
-This code defines a FastAPI endpoint at `/matrix_multiplication` that takes two matrices as input and returns the multiplication of the two matrices. The `np.matmul` function is used to perform the multiplication. The endpoint also includes a check to ensure that the number of columns in the first matrix is equal to the number of rows in the second matrix.
+The abbreviated terms in the given discharge summary that requires expansion are SARS-CoV-2, ARDS, ICU, SOEB, CPAx, and CPAx 6/50.
 
-To use this endpoint, you can make a POST request to `http://localhost:8000/matrix_multiplication` with the two matrices as input. The response will include the multiplication of the two matrices.
-```python
-import requests
-
-matrix1 = np.array([[1, 2], [3, 4]])
-matrix2 = np.array([[5, 6], [7, 8]])
-
-response = requests.post("http://localhost:8000/matrix_multiplication", json={"matrix1": matrix1, "matrix2": matrix2})
-
-print(response.json())
+### Response:
+The abbreviated terms in the given discharge summary that requires expansion are SARS-CoV-2, ARDS, ICU, SOEB, CPAx, and CPAx 6/50. SARS-CoV-2 stands for severe acute respiratory syndrome coronavirus 2, ARDS stands for acute respiratory distress syndrome, ICU stands for intensive care unit, SOEB stands for spontaneous breathing exercise, CPAx stands for Canadian Physical Activity Assessment, and CPAx 6/50 stands for a score of 6 out of 50 on the Canadian Physical Activity Assessment
 ```
-This code makes a POST request to the endpoint with the two matrices as input and prints the response. The response should include the multiplication of the two matrices, which is `[[11, 14], [29, 36]]`.
-
 ## License
 
 This model is released under the MIT License.