Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ index = faiss.read_index("index.faiss")
|
|
29 |
embedding_model = SentenceTransformer("sentence-transformers/paraphrase-multilingual-mpnet-base-v2")
|
30 |
|
31 |
# Load Jais model and tokenizer
|
32 |
-
model_name = "
|
33 |
tokenizer = AutoTokenizer.from_pretrained(model_name, token=HF_TOKEN, trust_remote_code=True)
|
34 |
model = AutoModelForCausalLM.from_pretrained(model_name, token=HF_TOKEN, trust_remote_code=True)
|
35 |
|
|
|
29 |
embedding_model = SentenceTransformer("sentence-transformers/paraphrase-multilingual-mpnet-base-v2")
|
30 |
|
31 |
# Load Jais model and tokenizer
|
32 |
+
model_name = "aubmindlab/aragpt2-base"
|
33 |
tokenizer = AutoTokenizer.from_pretrained(model_name, token=HF_TOKEN, trust_remote_code=True)
|
34 |
model = AutoModelForCausalLM.from_pretrained(model_name, token=HF_TOKEN, trust_remote_code=True)
|
35 |
|