connect model rag

rag.py

@@ -17,12 +17,12 @@ class RAG:
 
     # Download the vectorstore from Hugging Face Hub
     
-    def __init__(self, hf_token, embeddings_model, repo_name):
+    def __init__(self, hf_token, embeddings_model, repo_name,model_name):
 
         vectorstore = snapshot_download(repo_name)
 
 
-        # self.model_name = model_name
+        self.model_name = model_name
         self.hf_token = hf_token
         # self.rerank_model = rerank_model
         # self.rerank_number_contexts = rerank_number_contexts
@@ -153,8 +153,8 @@ class RAG:
 
             del model_parameters["NUM_CHUNKS"]
 
-            # response = self.predict_completion(prompt, text_context, model_parameters)
-            response = "Output"
+            response = self.predict_completion(prompt, text_context, model_parameters)
+            #response = "Output"
 
             if not response:
                 return self.NO_ANSWER_MESSAGE