ambrosfitz commited on
Commit
a52aae3
·
verified ·
1 Parent(s): dce04c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -29,7 +29,7 @@ HISTORY_INDEX = "onramps"
29
  # Model configurations
30
  MODELS = {
31
  "Cohere": {
32
- "name": "command-r-08-2024",
33
  "api_url": COHERE_API_URL,
34
  "api_key": COHERE_API_KEY
35
  },
@@ -93,7 +93,7 @@ def generate_answer_mistral(question, context, citations):
93
  "Accept": "application/json"
94
  }
95
 
96
- prompt = f"Context: {context}\n\nQuestion: {question}\n\nAnswer the question based on the given context. Include citations as [1], [2], etc.:"
97
 
98
  payload = {
99
  "model": MODELS['Mistral']['name'],
 
29
  # Model configurations
30
  MODELS = {
31
  "Cohere": {
32
+ "name": "command-r-plus-08-2024",
33
  "api_url": COHERE_API_URL,
34
  "api_key": COHERE_API_KEY
35
  },
 
93
  "Accept": "application/json"
94
  }
95
 
96
+ prompt = f"Context: {context}\n\nQuestion: {question}\n\nAnswer the question based on the given context and any pre-trained knowledge. Include citations as [1], [2], etc.:"
97
 
98
  payload = {
99
  "model": MODELS['Mistral']['name'],