Update djezzy.py
Browse files
djezzy.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
|
2 |
-
|
3 |
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig,HfArgumentParser,TrainingArguments,pipeline, logging
|
4 |
|
5 |
import os
|
@@ -51,7 +51,7 @@ def mot_cle(path):
|
|
51 |
|
52 |
|
53 |
|
54 |
-
def pip(question,docs_text, docs_embeddings):
|
55 |
query_text = question
|
56 |
query_embedding = embedding_llm.embed_query(query_text)
|
57 |
query_embedding_array = np.array(query_embedding)
|
|
|
1 |
|
2 |
+
|
3 |
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig,HfArgumentParser,TrainingArguments,pipeline, logging
|
4 |
|
5 |
import os
|
|
|
51 |
|
52 |
|
53 |
|
54 |
+
def pip(question,docs_text, docs_embeddings,mots_a_verifier):
|
55 |
query_text = question
|
56 |
query_embedding = embedding_llm.embed_query(query_text)
|
57 |
query_embedding_array = np.array(query_embedding)
|