acumplid commited on
Commit
359b142
1 Parent(s): 80ae396

Modified split

Browse files
Files changed (1) hide show
  1. rag.py +1 -1
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:-1]
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