saraht14 commited on
Commit
a290dce
·
verified ·
1 Parent(s): c5a68b2

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +19 -1
main.py CHANGED
@@ -4,4 +4,22 @@ app = Flask(__name__)
4
 
5
  @app.route("/")
6
  def hello():
7
- return jsonify({"status": "running"})
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
 
5
  @app.route("/")
6
  def hello():
7
+ return jsonify({"status": "running"})
8
+
9
+ from flask import Flask, jsonify, request
10
+ from datasets import load_dataset, Dataset
11
+ import pandas as pd
12
+ import os
13
+
14
+ HF_TOKEN = os.environ.get("HF_TOKEN")
15
+ REPO_ID = "saraht14/responses"
16
+
17
+ @app.route("/leaderboard", methods=["GET"])
18
+ def get_leaderboard():
19
+ try:
20
+ ds = load_dataset(REPO_ID, split="train", use_auth_token=HF_TOKEN)
21
+ df = ds.to_pandas()
22
+ df = df.sort_values(by=["Accuracy", "Execution Time (s)"], ascending=[False, True])
23
+ return jsonify(df.to_dict(orient="records"))
24
+ except Exception as e:
25
+ return jsonify({"error": f"Failed to load leaderboard: {str(e)}"}), 500