Update app.py

app.py

@@ -18,8 +18,8 @@ NUM_THREADS = int(os.getenv("NUM_THREADS"))
 
 print (NUM_THREADS)
 
-#@retry(stop=stop_after_attempt(5), wait=wait_fixed(2))
-@timeout_decorator.timeout(120)
+#@timeout_decorator.timeout(120)
+@retry(stop=stop_after_attempt(5), wait=wait_fixed(2))
 def call_openai_api(payload, headers):
     response = requests.post(API_URL, headers=headers, json=payload, stream=True)
     response.raise_for_status()