Update app.py

app.py

@@ -3,20 +3,34 @@ import spaces
 import subprocess
 import requests
 import datetime
+import os
 
-@spaces.GPU  
+@spaces.GPU
 def run_peft_eval():
-    subprocess.run(["python3", "run_eval.py"], check=True)
-    print("Evaluation complete.")
+    print("Starting evaluation via run_eval.py")
     try:
-        requests.post(os.environ["HF_ENDPOINT_SHUTDOWN"])  # Shutdown space after run
+        result = subprocess.run(["python3", "run_eval.py"], capture_output=True, text=True, check=True)
+        print(result.stdout)
+    except subprocess.CalledProcessError as e:
+        print("❌ Evaluation script failed:")
+        print(e.stderr)
+        return "Evaluation failed. Check logs."
+
+    print("Evaluation complete. Attempting shutdown.")
+    try:
+        shutdown_url = os.environ.get("HF_ENDPOINT_SHUTDOWN")
+        if shutdown_url:
+            requests.post(shutdown_url)
+        else:
+            print("No shutdown endpoint found.")
     except Exception as e:
         print("Failed to shutdown:", e)
-    return f"PEFT-Bench completed at {datetime.datetime.utcnow():%Y-%m-%d %H:%M UTC}"
+
+    return f"✅ PEFT-Bench completed at {datetime.datetime.utcnow():%Y-%m-%d %H:%M UTC}"
 
 with gr.Blocks() as demo:
-    gr.Markdown("### Running PEFT-Bench Evaluation...\nThis Space will shut down when done.")
-    output = gr.Textbox(label="Status")
+    gr.Markdown("Running PEFT-Bench Evaluation...\nThis Space will shut down when done.")
+    output = gr.Textbox(label="Status", lines=2)
     demo.load(fn=run_peft_eval, outputs=output)
 
 demo.launch()