PablitoGil14 commited on
Commit
ab52bf6
·
verified ·
1 Parent(s): d4fbf2b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ vectordb = Chroma.from_documents(splits, embedding=embeddings)
22
  ranker = Reranker("answerdotai/answerai-colbert-small-v1", model_type="colbert")
23
 
24
  # Cargar modelo de lenguaje de Hugging Face
25
- model_id = "mistralai/Mistral-7B-Instruct-v0.1"
26
  tokenizer = AutoTokenizer.from_pretrained(model_id)
27
  model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype="auto")
28
  generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
22
  ranker = Reranker("answerdotai/answerai-colbert-small-v1", model_type="colbert")
23
 
24
  # Cargar modelo de lenguaje de Hugging Face
25
+ model_id = "tiiuae/falcon-7b-instruct"
26
  tokenizer = AutoTokenizer.from_pretrained(model_id)
27
  model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype="auto")
28
  generator = pipeline("text-generation", model=model, tokenizer=tokenizer)