Update app.py

app.py

@@ -47,12 +47,10 @@ def invoke(openai_api_key, prompt, agent_option):
     
             result = completion["output"]
         elif (agent_option == AGENT_LLAMAINDEX):
-            print("111")
             result = agent_llamaindex(
                 config,
                 prompt
             )
-            print("222")
         else:
             client = OpenAI()
     
@@ -65,12 +63,10 @@ def invoke(openai_api_key, prompt, agent_option):
             callback = completion.usage
             result = completion.choices[0].message.content
     except Exception as e:
-        print("333")
         err_msg = e
 
         raise gr.Error(e)
     finally:
-        print("444")
         end_time_ms = round(time.time() * 1000)
         
         trace_wandb(
@@ -84,9 +80,7 @@ def invoke(openai_api_key, prompt, agent_option):
             start_time_ms, 
             end_time_ms
         )
-        print("555")
 
-    #print("666" + result)
     return result
 
 gr.close_all()