crodri commited on
Commit
490a51a
·
verified ·
1 Parent(s): 7603527

Update rag.py

Browse files
Files changed (1) hide show
  1. rag.py +1 -1
rag.py CHANGED
@@ -55,7 +55,7 @@ class RAG:
55
 
56
  payload = {
57
  "inputs": prompt + query,
58
- "parameters": {"MAX_NEW_TOKENS": 750, "TEMPERATURE": 0.35}
59
  }
60
 
61
  response = requests.post(self.model_name, headers=headers, json=payload)
 
55
 
56
  payload = {
57
  "inputs": prompt + query,
58
+ "parameters": {"MAX_NEW_TOKENS": 1000, "TEMPERATURE": 0.25}
59
  }
60
 
61
  response = requests.post(self.model_name, headers=headers, json=payload)