Spaces:

drvai
/

devai-demo

Runtime error

aakash0017 commited on Nov 10, 2023

Commit

11954cc

1 Parent(s): 78d0916

Minor change

Files changed (1) hide show

conversation.py CHANGED Viewed

@@ -146,8 +146,8 @@ def get_bert_embeddings(sentence):
     return embedding
 model_name = "BAAI/bge-base-en-v1.5"
-model = AutoModel.from_pretrained("models/models--BAAI--bge-base-en-v1.5/snapshots/617ca489d9e86b49b8167676d8220688b99db36e")
-tokenizer = AutoTokenizer.from_pretrained("models/models--BAAI--bge-base-en-v1.5/snapshots/617ca489d9e86b49b8167676d8220688b99db36e")
 prompt_file = open("prompts/version_2.txt", "r").read()
 pinecone.init(

     return embedding
 model_name = "BAAI/bge-base-en-v1.5"
+model = AutoModel.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
 prompt_file = open("prompts/version_2.txt", "r").read()
 pinecone.init(