Update app.py

app.py

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