Spaces:
Running
Running
Update rag.py
Browse files
rag.py
CHANGED
@@ -55,7 +55,7 @@ class RAG:
|
|
55 |
|
56 |
payload = {
|
57 |
"inputs": prompt + query,
|
58 |
-
"parameters": {"MAX_NEW_TOKENS":
|
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)
|