Spaces:
Runtime error
Runtime error
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)
|