Update app.py

app.py

@@ -156,7 +156,8 @@ with gr.Blocks() as demo:
             
             start_time = time.time()
             keep_streaming = True
-            with requests.post(API_URL, headers=HEADERS, json=data, stream=True, timeout=30) as r:
+            s = requests.Session()
+            with s.post(API_URL, headers=HEADERS, json=data, stream=True) as r:
                 for line in r.iter_lines():
                     if line and keep_streaming:
                         assert r.status_code == 200