Joschka Strueber commited on
Commit
4b2993a
·
1 Parent(s): 75132dc

[Fix, Debug] wrong default model, check filter_labels

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. src/dataloading.py +1 -0
app.py CHANGED
@@ -98,7 +98,7 @@ links_markdown = """
98
  [🤗 Data](https://huggingface.co/datasets/bethgelab/lm-similarity)
99
  """
100
 
101
- model_init = ["HuggingFaceTB/SmolLM2-1.7B-Instruct", "meta-llama/Llama-3.1-8B-Instruct", "microsoft/phi-4", "google/gemma-2-27b-it", "Qwen/Qwen2.5-32B-Instruct", "meta-llama/Llama-3.1-8B-Instruct"]
102
 
103
  # Create Gradio interface
104
  with gr.Blocks(title="LLM Similarity Analyzer") as demo:
 
98
  [🤗 Data](https://huggingface.co/datasets/bethgelab/lm-similarity)
99
  """
100
 
101
+ model_init = ["HuggingFaceTB/SmolLM2-1.7B-Instruct", "meta-llama/Llama-3.1-8B-Instruct", "microsoft/phi-4", "google/gemma-2-27b-it", "Qwen/Qwen2.5-32B-Instruct", "meta-llama/Llama-3.3-70B-Instruct"]
102
 
103
  # Create Gradio interface
104
  with gr.Blocks(title="LLM Similarity Analyzer") as demo:
src/dataloading.py CHANGED
@@ -95,6 +95,7 @@ def filter_labels(doc):
95
  for d in doc:
96
  labels.append(d["answer_index"])
97
  else:
 
98
  for d in doc:
99
  if d["answer"] == "False":
100
  labels.append(0)
 
95
  for d in doc:
96
  labels.append(d["answer_index"])
97
  else:
98
+ print("Keys:", doc[0].keys())
99
  for d in doc:
100
  if d["answer"] == "False":
101
  labels.append(0)