Update app.py

app.py

@@ -34,6 +34,24 @@ function refresh() {
 
 previous_sessions = []
 
+def post_process_output(prompt, result):
+    # Remove the prompt if it's repeated at the beginning of the answer
+    answer = result.strip()
+    if answer.startswith(prompt):
+        answer = answer[len(prompt):].strip()
+
+    # Remove any leading colons or whitespace
+    answer = answer.lstrip(':')
+
+    # Ensure the answer starts with a capital letter
+    answer = answer.capitalize()
+
+    # Ensure the answer ends with a period if it doesn't already
+    if not answer.endswith('.'):
+        answer += '.'
+
+    return f"{answer}"
+    
 
 # def add_session(prompt):
 #     global previous_sessions
@@ -46,27 +64,27 @@ previous_sessions = []
 
 
 
-# def inference(prompt):
-#   prompt_text = prompt
-#   generated_text = gemma_lm.generate(prompt_text)
+def inference(prompt):
+  prompt_text = prompt
+  generated_text = gemma_lm.generate(prompt_text)
 
-#   #Apply post-processing
-#   formatted_output = post_process_output(prompt_text, generated_text)
-#   print(formatted_output)
+  #Apply post-processing
+  formatted_output = post_process_output(prompt_text, generated_text)
+  print(formatted_output)
 
-#   #adding a bit of delay
-#   time.sleep(1)
-#   result = formatted_output
-#   sessions = add_session(prompt_text)
-#   return result, sessions
+  #adding a bit of delay
+  time.sleep(1)
+  result = formatted_output
+  # sessions = add_session(prompt_text)
+  return result
 
 
-def inference(prompt):
+# def inference(prompt):
     
-    time.sleep(1)
-    result = "Your Result"
-    # sessions = add_session(prompt)
-    return result
+#     time.sleep(1)
+#     result = "Your Result"
+#     # sessions = add_session(prompt)
+#     return result
 
      
 # # def remember(prompt, result):