Update app.py

app.py

@@ -106,11 +106,12 @@ 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)
+            match12 = re.search(r'Solution:(.*?)', solution_text, re.DOTALL)
             if match12:
                 solution_text=match12.group(1).strip()
                 print(solution_text)