Spaces:

ccm
/

chat-with-publications

Running on Zero

ccm commited on Jul 25

Commit

a1a6086

•

1 Parent(s): 5e12e86

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -24,7 +24,7 @@ EXAMPLE_QUERIES = [
     "What can agent-based models do about climate change?",
 ]
 EMBEDDING_MODEL_NAME = "allenai-specter"
-LLM_MODEL_NAME = "Qwen/Qwen2-1.5B-Instruct"
 # Load the dataset and convert to pandas
 data = datasets.load_dataset("ccm/publications")["train"].to_pandas()

     "What can agent-based models do about climate change?",
 ]
 EMBEDDING_MODEL_NAME = "allenai-specter"
+LLM_MODEL_NAME = "Qwen/Qwen2-7B-Instruct"
 # Load the dataset and convert to pandas
 data = datasets.load_dataset("ccm/publications")["train"].to_pandas()