Spaces:
Sleeping
Sleeping
acumplid
commited on
Commit
·
359b142
1
Parent(s):
80ae396
Modified split
Browse files
rag.py
CHANGED
|
@@ -52,7 +52,7 @@ class RAG:
|
|
| 52 |
|
| 53 |
response = requests.post(self.model_name, headers=headers, json=payload)
|
| 54 |
|
| 55 |
-
return response.json()[0]["generated_text"].split("###")[-1][8
|
| 56 |
|
| 57 |
def beautiful_context(self, docs):
|
| 58 |
|
|
|
|
| 52 |
|
| 53 |
response = requests.post(self.model_name, headers=headers, json=payload)
|
| 54 |
|
| 55 |
+
return response.json()[0]["generated_text"].split("###")[-1][8:]
|
| 56 |
|
| 57 |
def beautiful_context(self, docs):
|
| 58 |
|