Spaces:
Running
Running
Joschka Strueber
commited on
Commit
·
4b2993a
1
Parent(s):
75132dc
[Fix, Debug] wrong default model, check filter_labels
Browse files- app.py +1 -1
- 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.
|
| 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)
|