Update gradio_app.py

gradio_app.py

@@ -1,5 +1,6 @@
 # Importing libraries
 from llama_cpp import Llama
+from time import time
 import gradio as gr
 import psutil
 
@@ -30,6 +31,7 @@ with open('system.message', 'r', encoding='utf-8') as f:
     system_message = f.read()
 
 def generate_answer(request: str, max_tokens: int = 256, custom_prompt: str = None):
+    t0 = time()
     logs = f"Request: {request}\nMax tokens: {max_tokens}\nCustom prompt: {custom_prompt}\n"
     try:
         maxTokens = max_tokens if 16 <= max_tokens <= 256 else 64
@@ -58,11 +60,12 @@ def generate_answer(request: str, max_tokens: int = 256, custom_prompt: str = No
             text = "Sorry, but something went wrong while generating answer. Try again or fix code. If you are maintainer of this space, look into logs."
         
         logs += f"\nFinal: '''{text}'''"
+        logs += f"\n\nTime spent: {time()-t0}"
         return text, logs
     except Exception as e:
-        print(e)
+        logs += str(e)
+        logs += f"\n\nTime spent: {time()-t0}"
         return "Oops! Internal server error. Check the logs of space/instance.", logs
-    print("\n\n\n")
 
 print("! LOAD GRADIO INTERFACE !")
 demo = gr.Interface(