Update app.py

app.py

@@ -10,7 +10,7 @@ from langchain.llms import HuggingFacePipeline
 from langchain.memory import ConversationBufferMemory
 
 from transformers import AutoTokenizer, pipeline
-import chromadb
+import torch
 
 # Lista de modelos 100% abertos e gratuitos
 list_llm = [
@@ -125,15 +125,15 @@ def demo():
             outputs=[vector_db, process_status]
         )
         
-        def load_model(model, temp):
-            if vector_db.value is None:
+        def load_model(model, temp, vector_db_state):
+            if vector_db_state is None:
                 return None, "Por favor, processe os documentos primeiro."
-            qa = initialize_llmchain(list_llm[list_llm_simple.index(model)], temp, 512, 3, vector_db.value)
+            qa = initialize_llmchain(list_llm[list_llm_simple.index(model)], temp, 512, 3, vector_db_state)
             return qa, "Modelo carregado!"
         
         load_model_btn.click(
             load_model,
-            inputs=[model_selector, temperature],
+            inputs=[model_selector, temperature, vector_db],
             outputs=[qa_chain, model_status]
         )