Spaces:
Running
Running
Update rag.py
Browse files
rag.py
CHANGED
@@ -17,7 +17,6 @@ class RAG:
|
|
17 |
|
18 |
def __init__(self, hf_token, embeddings_model, model_name):
|
19 |
|
20 |
-
|
21 |
self.model_name = model_name
|
22 |
self.hf_token = hf_token
|
23 |
|
@@ -54,7 +53,7 @@ class RAG:
|
|
54 |
],
|
55 |
stream=False
|
56 |
)
|
57 |
-
|
58 |
return(chat_completion)
|
59 |
|
60 |
|
|
|
17 |
|
18 |
def __init__(self, hf_token, embeddings_model, model_name):
|
19 |
|
|
|
20 |
self.model_name = model_name
|
21 |
self.hf_token = hf_token
|
22 |
|
|
|
53 |
],
|
54 |
stream=False
|
55 |
)
|
56 |
+
print(chat_completion)
|
57 |
return(chat_completion)
|
58 |
|
59 |
|