Update app.py

app.py

@@ -61,10 +61,7 @@ print("Finish the model init process")
 
 def qa(question: str) -> str:
     
-    vectordb = Chroma(
-                    persist_directory="./resource/chroma/",
-                    embedding_function=embeddings
-                )
+    
     docs = vectordb.max_marginal_relevance_search(question,k=3)
 
     
@@ -81,7 +78,7 @@ def qa(question: str) -> str:
             #    """
 
     #outputs = model.generate(prompt=prompt, temp=0.5, top_k = 40, top_p = 1, max_tokens = max_new_tokens)
-    return docs #"".join(outputs)
+    return vectordb._collection.count() #"".join(outputs)
 
 
 demo = gr.Interface(