benchaffe commited on
Commit
216ff08
·
verified ·
1 Parent(s): 2ba4925

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -9
app.py CHANGED
@@ -32,20 +32,19 @@ def log_interaction(input_text, prediction):
32
  def predict_and_log(text):
33
  result = model(text)
34
  log_interaction(text, result)
35
- return result, download_file.update(value=LOG_FILE)
 
 
 
36
 
37
  with gr.Blocks() as demo:
38
  gr.Markdown("## Biomedical Abbreviation Identifier")
39
-
40
  input_box = gr.Textbox(label="Enter biomedical text")
41
  output_box = gr.JSON(label="Model Prediction")
42
- download_file = gr.File(label="Download Logs", value=LOG_FILE)
43
-
44
  submit_btn = gr.Button("Submit")
45
- submit_btn.click(
46
- fn=predict_and_log,
47
- inputs=input_box,
48
- outputs=[output_box, download_file]
49
- )
50
 
51
  demo.launch()
 
32
  def predict_and_log(text):
33
  result = model(text)
34
  log_interaction(text, result)
35
+ return result
36
+
37
+ def get_log_file():
38
+ return LOG_FILE
39
 
40
  with gr.Blocks() as demo:
41
  gr.Markdown("## Biomedical Abbreviation Identifier")
 
42
  input_box = gr.Textbox(label="Enter biomedical text")
43
  output_box = gr.JSON(label="Model Prediction")
44
+ download_output = gr.File(label="Click to download")
 
45
  submit_btn = gr.Button("Submit")
46
+ download_btn = gr.Button("Download Log")
47
+ submit_btn.click(fn=predict_and_log, inputs=input_box, outputs=output_box)
48
+ download_btn.click(fn=get_log_file, outputs=download_output)
 
 
49
 
50
  demo.launch()