TI_RAG_Demo_L3.1

Sleeping

arjunanand13 commited on Jul 1

Commit

b4efea2

•

1 Parent(s): 34bed84

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -106,7 +106,7 @@ class DocumentRetrievalAndGeneration:
         decoded = self.llm.tokenizer.batch_decode(generated_ids)
         generated_response = decoded[0]
         match1 = re.search(r'\[/INST\](.*?)</s>', generated_response, re.DOTALL)
-        match2 = re.search(r'Solution:(.*?)</s>', text, re.DOTALL | re.IGNORECASE)
         if match1:
             solution_text = match1.group(1).strip()
             print(solution_text)

         decoded = self.llm.tokenizer.batch_decode(generated_ids)
         generated_response = decoded[0]
         match1 = re.search(r'\[/INST\](.*?)</s>', generated_response, re.DOTALL)
+        match2 = re.search(r'Solution:(.*?)</s>', generated_response, re.DOTALL | re.IGNORECASE)
         if match1:
             solution_text = match1.group(1).strip()
             print(solution_text)