Update app.py

app.py

@@ -245,7 +245,7 @@ def predict(user_input, timeout_seconds=1800): # 30 minutes = 1800 seconds
     # while writing to the log file, ensure that the commit scheduler is locked to avoid parallel
     # access
 
-    return prediction, context_for_query, user_input
+    return prediction
 
 # Set-up the Gradio UI
 # Add text box.
@@ -254,7 +254,7 @@ textbox = gr.Textbox(placeholder="Enter your query here", lines=6)
 # Create the interface
 demo = gr.Interface(
     inputs=[textbox], fn=predict, outputs="text",
-    title="I Mat - Your Mental Trainer",
+    title="I Mat - Your Personal Mental Trainer",
     description="This web API presents an interface to interact with your mental trainer Mat",
     concurrency_limit=16
 )