Update app.py

app.py

@@ -2,16 +2,13 @@ import gradio as gr
 import logging, os, sys, threading, time
 
 from agent_langchain import agent_langchain
-from agent_llamaindex import agent_llamaindex
+#from agent_llamaindex import agent_llamaindex
 from openai import OpenAI
 from trace import trace_wandb
 
 from dotenv import load_dotenv, find_dotenv
 _ = load_dotenv(find_dotenv())
 
-import nltk
-nltk.download('wordnet')
-
 lock = threading.Lock()
 
 AGENT_OFF = "Off"
@@ -52,11 +49,11 @@ def invoke(openai_api_key, prompt, agent_option):
                 )
         
                 result = completion["output"]
-            elif (agent_option == AGENT_LLAMAINDEX):
-                result = agent_llamaindex(
-                    config,
-                    prompt
-                )
+            #elif (agent_option == AGENT_LLAMAINDEX):
+            #    result = agent_llamaindex(
+            #        config,
+            #        prompt
+            #    )
             else:
                 client = OpenAI()