Hugging Face's logo

Spaces:
tongyi21
/
AI_Novelist_RAG
Sleeping

App Files Community
nanfangwuyu21 commited on
Commit
1f16680
·
1 Parent(s): 847f4ee

change summary model to openai gpt-4o-mini

Browse files
Files changed (3) hide show
  1. README.md +1 -0
  2. app/config.py +2 -0
  3. app/models/model.py +2 -1
README.md CHANGED
@@ -7,6 +7,7 @@ sdk: docker
7
  sdk_version: 28.0.4
8
  python_version: 3.10.16
9
  app_port: 7860
 
10
  ---
11
 
12
 
 
7
  sdk_version: 28.0.4
8
  python_version: 3.10.16
9
  app_port: 7860
10
+ accelerator: cpu # gpu when local, be online in future
11
  ---
12
 
13
 
app/config.py CHANGED
@@ -0,0 +1,2 @@
 
 
 
1
+ Chapter_LLM = "gpt-4o-mini"
2
+ Summary_LLM = "gpt-4o-mini" # "bart-large-cnn"
app/models/model.py CHANGED
@@ -5,5 +5,6 @@ from langchain_huggingface import HuggingFaceEmbeddings
5
 
6
  # model = TinyLlamaModel()
7
  LLM = GPT4OMini()
8
- Summary_Model = BartSummaryModel()
 
9
  Embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
 
5
 
6
  # model = TinyLlamaModel()
7
  LLM = GPT4OMini()
8
+ # Summary_Model = BartSummaryModel()
9
+ Summary_Model = GPT4OMini()
10
  Embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")