Update app.py

app.py

@@ -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)
+        match2 = re.search(r'Solution:(.*?)</s>', generated_response, re.DOTALL | re.IGNORECASE)
         if match1:
             solution_text = match1.group(1).strip()
             print(solution_text)