Mdrnfox commited on
Commit
2d819a2
·
verified ·
1 Parent(s): 582209c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -10
app.py CHANGED
@@ -7,30 +7,29 @@ import os
7
 
8
  @spaces.GPU
9
  def run_peft_eval():
10
- print("Starting evaluation via run_eval.py")
11
  try:
12
  result = subprocess.run(["python3", "run_eval.py"], capture_output=True, text=True, check=True)
13
- print(result.stdout)
14
  except subprocess.CalledProcessError as e:
15
- print("Evaluation script failed:")
16
- print(e.stderr)
17
  return "Evaluation failed. Check logs."
18
 
19
- print("Evaluation complete. Attempting shutdown.")
20
  try:
21
  shutdown_url = os.environ.get("HF_ENDPOINT_SHUTDOWN")
22
  if shutdown_url:
 
23
  requests.post(shutdown_url)
24
- else:
25
- print("No shutdown endpoint found.")
26
  except Exception as e:
27
- print("Failed to shutdown:", e)
28
 
29
  return f"✅ PEFT-Bench completed at {datetime.datetime.utcnow():%Y-%m-%d %H:%M UTC}"
30
 
31
  with gr.Blocks() as demo:
32
- gr.Markdown("Running PEFT-Bench Evaluation...\nThis Space will shut down when done.")
 
33
  output = gr.Textbox(label="Status", lines=2)
34
- demo.load(fn=run_peft_eval, outputs=output)
35
 
36
  demo.launch()
 
7
 
8
  @spaces.GPU
9
  def run_peft_eval():
10
+ print("Starting evaluation via run_eval.py", flush=True)
11
  try:
12
  result = subprocess.run(["python3", "run_eval.py"], capture_output=True, text=True, check=True)
13
+ print(result.stdout, flush=True)
14
  except subprocess.CalledProcessError as e:
15
+ print("Evaluation script failed:", flush=True)
16
+ print(e.stderr, flush=True)
17
  return "Evaluation failed. Check logs."
18
 
 
19
  try:
20
  shutdown_url = os.environ.get("HF_ENDPOINT_SHUTDOWN")
21
  if shutdown_url:
22
+ print("Evaluation done. Shutting down...", flush=True)
23
  requests.post(shutdown_url)
 
 
24
  except Exception as e:
25
+ print("Failed to shutdown:", e, flush=True)
26
 
27
  return f"✅ PEFT-Bench completed at {datetime.datetime.utcnow():%Y-%m-%d %H:%M UTC}"
28
 
29
  with gr.Blocks() as demo:
30
+ gr.Markdown("## PEFT-Bench Evaluation")
31
+ gr.Markdown("This Space runs evaluation and automatically shuts down when done.")
32
  output = gr.Textbox(label="Status", lines=2)
33
+ demo.load(fn=run_peft_eval, outputs=output, every=0.1)
34
 
35
  demo.launch()