MuXodious commited on
Commit
6a3fafe
·
1 Parent(s): 29c5bd3

Address an edge case with KLD parsing

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. src/streamlit_app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ def fetch_model_info(model):
33
 
34
  # Extract KL divergence using Regex
35
  # Matches formats like: "KL divergence | 0.0033"
36
- kl_match = re.search(r"(?i)KL\s*divergence[^\d\n]*?([\d\.]+)", readme_text)
37
  if kl_match:
38
  try:
39
  kl_div = float(kl_match.group(1))
 
33
 
34
  # Extract KL divergence using Regex
35
  # Matches formats like: "KL divergence | 0.0033"
36
+ kl_match = re.search(r"(?i)KL\s*divergence[^a-zA-Z\d\n]*?([\d\.]+)", readme_text)
37
  if kl_match:
38
  try:
39
  kl_div = float(kl_match.group(1))
src/streamlit_app.py CHANGED
@@ -33,7 +33,7 @@ def fetch_model_info(model):
33
 
34
  # Extract KL divergence using Regex
35
  # Matches formats like: "KL divergence | 0.0033"
36
- kl_match = re.search(r"(?i)KL\s*divergence[^\d\n]*?([\d\.]+)", readme_text)
37
  if kl_match:
38
  try:
39
  kl_div = float(kl_match.group(1))
 
33
 
34
  # Extract KL divergence using Regex
35
  # Matches formats like: "KL divergence | 0.0033"
36
+ kl_match = re.search(r"(?i)KL\s*divergence[^a-zA-Z\d\n]*?([\d\.]+)", readme_text)
37
  if kl_match:
38
  try:
39
  kl_div = float(kl_match.group(1))