Download src/postprocessing.py from LabSoft/Epikriza: direct link, hf CLI and curl.
- Browser
- Download file 861 Bytes
-
https://huggingface.co/spaces/LabSoft/Epikriza/resolve/main/src/postprocessing.py
- Command line
-
hf download hf://spaces/LabSoft/Epikriza/src/postprocessing.py
-
curl -L -o postprocessing.py https://huggingface.co/spaces/LabSoft/Epikriza/resolve/main/src/postprocessing.py
861 Bytes
| import openai | |
| from dotenv import load_dotenv | |
| load_dotenv() | |
| POSTROCESSINGPROMPT = """Ti si specijalizovan za medicinsku komunikaciju. Tvoj zadatak je da gramatički ispraviš tekst bez menjanja originalnog teksta. Dodatno formatiraj lekarsku epikrizu tako da bude čitljiviji. Dodaj novi red gde je potrebno kako bi se poboljšala čitljivost. Lekarska epikriza sastoji se od stručnih medicinskih termina, tako da uzmi to u obzir prilikom leksičke analize i ispravljanja netačnih termina.""" | |
| def postprocessing(old_epikriza): | |
| response = openai.chat.completions.create( | |
| model="gpt-3.5-turbo-0125", | |
| messages=[ | |
| {"role": "system", "content": POSTROCESSINGPROMPT}, | |
| {"role": "user", "content": old_epikriza}, | |
| ], | |
| ) | |
| epikriza = response.choices[0].message.content | |
| print(epikriza) | |
| return epikriza | |