thecuong commited on
Commit
466646a
·
1 Parent(s): b0926d9

feat: update trust_remote_code=True

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,7 +2,8 @@ import gradio as gr
2
  from sentence_transformers import SentenceTransformer
3
 
4
  # Tải mô hình
5
- model = SentenceTransformer('Alibaba-NLP/gte-multilingual-base')
 
6
 
7
  def gte_model(sentences: list):
8
  try:
 
2
  from sentence_transformers import SentenceTransformer
3
 
4
  # Tải mô hình
5
+ model = SentenceTransformer(model_name_or_path='Alibaba-NLP/gte-multilingual-base',
6
+ trust_remote_code=True)
7
 
8
  def gte_model(sentences: list):
9
  try: