bulubula commited on
Commit
858de24
1 Parent(s): aecc216

model link

Browse files
Files changed (1) hide show
  1. utils.py +8 -0
utils.py CHANGED
@@ -45,6 +45,13 @@ def fetch_models_evaluation(api, projects, selected_project):
45
  try:
46
  summary = run.summary
47
  if summary.get("score") is not None: # Assuming runs with score are model evaluations
 
 
 
 
 
 
 
48
  data.append({
49
  "Created At": run.created_at,
50
 
@@ -60,6 +67,7 @@ def fetch_models_evaluation(api, projects, selected_project):
60
  "ROC AUC": summary.get("roc_auc"),
61
  "Confusion Matrix": summary.get("confusion_matrix"),
62
 
 
63
  "Score": summary.get("score"),
64
  #TODO link to huggingface model
65
  })
 
45
  try:
46
  summary = run.summary
47
  if summary.get("score") is not None: # Assuming runs with score are model evaluations
48
+ model_link = summary.get("model_link", "")
49
+ if model_link:
50
+ # Create clickable link
51
+ model_link_html = f'<a href="{model_link}" target="_blank">Model Link</a>'
52
+ else:
53
+ model_link_html = "N/A"
54
+
55
  data.append({
56
  "Created At": run.created_at,
57
 
 
67
  "ROC AUC": summary.get("roc_auc"),
68
  "Confusion Matrix": summary.get("confusion_matrix"),
69
 
70
+ "Model link": model_link_html,
71
  "Score": summary.get("score"),
72
  #TODO link to huggingface model
73
  })