Spaces:

MariiaRazno
/

ICD_DIA_extraction

Runtime error

MariiaRazno commited on Jun 6, 2023

Commit

3c2b1e2

1 Parent(s): 82a1932

output changes

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ def get_result(text):
     try:
         response_icd = GPT_answer(prompt_icd)
         response_dia = GPT_answer(prompt_dia)
-        answer = 'ICD-10 codes:'+response_icd+'\nFound diagnoses:\n'+response_dia
         return answer
     except openai.error.InvalidRequestError as e:
         # Handle the InvalidRequestError

     try:
         response_icd = GPT_answer(prompt_icd)
         response_dia = GPT_answer(prompt_dia)
+        answer = 'ICD-10 CODES:\n'+re.sub('\n', '', response_icd)+'\nFound DIAGNOSES:\n'+re.sub('\n', '', response_dia)
         return answer
     except openai.error.InvalidRequestError as e:
         # Handle the InvalidRequestError