ohalkhateeb commited on
Commit
25daef8
·
verified ·
1 Parent(s): c54a9b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "inceptionai/jais-13b"
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