rshakked commited on
Commit
c8d8b34
Β·
1 Parent(s): 96bc19d

fix: resolve model detection and evaluation dropdown issues in app

Browse files
Files changed (2) hide show
  1. app.py +5 -3
  2. train_abuse_model.py +0 -3
app.py CHANGED
@@ -36,11 +36,10 @@ with gr.Blocks() as demo:
36
  outputs=[enriched_output, label_output]
37
  )
38
 
39
-
40
  with gr.Tab("πŸ“Š View Evaluation Reports"):
41
  def list_eval_files():
42
  folder = "/home/user/app/results_eval"
43
- return sorted(os.listdir(folder), reverse=True)
44
 
45
  def load_eval_file(filename):
46
  path = f"/home/user/app/results_eval/{filename}"
@@ -50,15 +49,18 @@ with gr.Blocks() as demo:
50
  return f.read()
51
 
52
  file_dropdown = gr.Dropdown(
53
- choices=list_eval_files(),
54
  label="πŸ“ Select an evaluation file",
55
  interactive=True
56
  )
 
57
  report_output = gr.Textbox(label="πŸ“„ Evaluation Report", lines=20)
58
 
 
59
  file_dropdown.change(fn=load_eval_file, inputs=file_dropdown, outputs=report_output)
60
 
61
 
62
 
 
63
  if __name__ == "__main__":
64
  demo.launch()
 
36
  outputs=[enriched_output, label_output]
37
  )
38
 
 
39
  with gr.Tab("πŸ“Š View Evaluation Reports"):
40
  def list_eval_files():
41
  folder = "/home/user/app/results_eval"
42
+ return sorted(os.listdir(folder), reverse=True) if os.path.exists(folder) else []
43
 
44
  def load_eval_file(filename):
45
  path = f"/home/user/app/results_eval/{filename}"
 
49
  return f.read()
50
 
51
  file_dropdown = gr.Dropdown(
52
+ choices=[],
53
  label="πŸ“ Select an evaluation file",
54
  interactive=True
55
  )
56
+ refresh_btn = gr.Button("πŸ”„ Refresh File List")
57
  report_output = gr.Textbox(label="πŸ“„ Evaluation Report", lines=20)
58
 
59
+ refresh_btn.click(fn=lambda: gr.Dropdown.update(choices=list_eval_files()), outputs=file_dropdown)
60
  file_dropdown.change(fn=load_eval_file, inputs=file_dropdown, outputs=report_output)
61
 
62
 
63
 
64
+
65
  if __name__ == "__main__":
66
  demo.launch()
train_abuse_model.py CHANGED
@@ -300,9 +300,6 @@ def run_training(progress=gr.Progress(track_tqdm=True)):
300
  progress(1.0)
301
  yield "βœ… Progress: 100%\n"
302
 
303
- PERSIST_DIR = Path("/home/user/app")
304
- MODEL_DIR = PERSIST_DIR / "saved_model"
305
-
306
  # Save the model and tokenizer
307
  MODEL_DIR.mkdir(parents=True, exist_ok=True)
308
  model.save_pretrained(MODEL_DIR)
 
300
  progress(1.0)
301
  yield "βœ… Progress: 100%\n"
302
 
 
 
 
303
  # Save the model and tokenizer
304
  MODEL_DIR.mkdir(parents=True, exist_ok=True)
305
  model.save_pretrained(MODEL_DIR)