Upload app.py

app.py

@@ -5,6 +5,8 @@ import os
 import gradio as gr
 
 ML_ENDPOINT_URL = os.environ.get("ML_ENDPOINT_URL", "http://50.18.255.74:8040/rewrite")
+client_session = requests.Session()
+client_session.keep_alive = 5
 
 # Example queries to help users understand the app
 EXAMPLE_QUERIES = [
@@ -27,7 +29,7 @@ def make_request(query):
         # Replace with your actual API endpoint
         if "rewrite: " not in query:
             query = f"rewrite: {query}"
-        response = requests.post(ML_ENDPOINT_URL, json={"inputs": query})
+        response = client_session.post(ML_ENDPOINT_URL, json={"inputs": query})
         response.raise_for_status()
         result = response.json()
     except requests.exceptions.RequestException as e: