Spaces:
Running
on
Zero
Running
on
Zero
Update main.py
Browse files
main.py
CHANGED
@@ -26,7 +26,7 @@ EXAMPLE_QUERIES = [
|
|
26 |
"What is known about different modes for human-AI teaming?",
|
27 |
]
|
28 |
EMBEDDING_MODEL_NAME = "allenai-specter"
|
29 |
-
LLM_MODEL_NAME = "Qwen/Qwen2-7B-Instruct"
|
30 |
|
31 |
# Load the dataset and convert to pandas
|
32 |
data = pandas.read_parquet("hf://datasets/ccm/publications/data/train-00000-of-00001.parquet")
|
|
|
26 |
"What is known about different modes for human-AI teaming?",
|
27 |
]
|
28 |
EMBEDDING_MODEL_NAME = "allenai-specter"
|
29 |
+
LLM_MODEL_NAME = "Qwen/Qwen2-7B-Instruct-AWQ"
|
30 |
|
31 |
# Load the dataset and convert to pandas
|
32 |
data = pandas.read_parquet("hf://datasets/ccm/publications/data/train-00000-of-00001.parquet")
|