Update app.py

app.py

@@ -26,10 +26,10 @@ api=HfApi()
 
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
+client_sort = Client("Omnibus/sort_document")
 
 def sort_fn(inp):
 
-    client_sort = Client("Omnibus/sort_document")
     sen,nouns = client_sort.predict(
     		f"{inp}",	# str  in 'Paste Text' Textbox component
     		api_name="/sort_doc"
@@ -39,8 +39,7 @@ def sort_fn(inp):
 
 
 def find_fn(inp,sen,nouns):
-    client = Client("Omnibus/sort_document")
-    result = client.predict(
+    result = client_sort.predict(
     		inp,	# str  in 'Search query' Textbox component
     		sen,	# Dict[Any, Any] (any valid json) in 'Sentences' Json component
     		nouns,	# Dict[Any, Any] (any valid json) in 'Nouns' Json component
@@ -488,7 +487,7 @@ def search_fn(inp,history,report_check,chart_check,data):
         inp = "Process this data"
     history.clear()
     history = [(inp,"Working on it...")] 
-    yield "",history,error_box,json_box,json_box
+    yield "",history,error_box,None,None
 
     
     if data != "Error" and data != "":