Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 = "
|
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)
|