imessien commited on
Commit
f71ea87
·
1 Parent(s): 6c9bb82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,5 +14,5 @@ def prompt(text):
14
  return [ config.id2label[ids] for ids in results]
15
  #return tokenizer.decode(output[0][0].argmax(dim=-1))
16
 
17
- gr.Interface.load(fn=prompt, inputs="text", outputs="text",
18
- title="ICD 10 Code Prediction", description="type in a disease and it will predict its ICD-10 codes",).launch(debug=True)
 
14
  return [ config.id2label[ids] for ids in results]
15
  #return tokenizer.decode(output[0][0].argmax(dim=-1))
16
 
17
+ demo=gr.Interface.load(fn=prompt, inputs="text", outputs="text",
18
+ title="ICD 10 Code Prediction", description="type in a disease and it will predict its ICD-10 codes")