Update app.py

app.py

@@ -28,6 +28,21 @@ mergedModel.eval()
 
 
 
+def extract_answer(message):
+    # Find the index of '### Answer:'
+    start_index = message.find('### Answer:')
+    if start_index != -1:
+        # Extract the part of the message after '### Answer:'
+        answer_part = message[start_index + len('### Answer:'):].strip()
+        # Find the index of the last full stop
+        last_full_stop_index = answer_part.rfind('.')
+        if last_full_stop_index != -1:
+            # Remove the part after the last full stop
+            answer_part = answer_part[:last_full_stop_index + 1]
+        return answer_part.strip()  # Remove leading and trailing whitespace
+    else:
+        return "I don't have the answer to this question....."
+
 def inferance(query: str, model, tokenizer, temp = 1.0, limit = 200) -> str:
   device = "cuda:0"
 
@@ -50,9 +65,12 @@ def inferance(query: str, model, tokenizer, temp = 1.0, limit = 200) -> str:
 
 
 
+
+
 def predict(temp, limit, text):
     prompt = text
     out = inferance(prompt, mergedModel, tokenizer, temp = 1.0, limit = 200)
+    display = extract_answer(out)
     return out
 
 pred = gr.Interface(