Update app.py
Browse files
app.py
CHANGED
@@ -56,9 +56,7 @@ class CFG:
|
|
56 |
Embeddings_path = '/content/faiss-hp-sentence-transformers'
|
57 |
Output_folder = './rag-vectordb'
|
58 |
|
59 |
-
|
60 |
-
|
61 |
-
print('\nDownloading model: ', model, '\n\n')
|
62 |
|
63 |
if model == 'wizardlm':
|
64 |
model_repo = 'TheBloke/wizardLM-7B-HF'
|
@@ -155,7 +153,6 @@ class CFG:
|
|
155 |
|
156 |
def get_model(model = CFG.model_name):
|
157 |
|
158 |
-
print('\nDownloading model: ', model, '\n\n')
|
159 |
|
160 |
if model == 'wizardlm':
|
161 |
model_repo = 'TheBloke/wizardLM-7B-HF'
|
|
|
56 |
Embeddings_path = '/content/faiss-hp-sentence-transformers'
|
57 |
Output_folder = './rag-vectordb'
|
58 |
|
59 |
+
def get_model(model = CFG.model_name):
|
|
|
|
|
60 |
|
61 |
if model == 'wizardlm':
|
62 |
model_repo = 'TheBloke/wizardLM-7B-HF'
|
|
|
153 |
|
154 |
def get_model(model = CFG.model_name):
|
155 |
|
|
|
156 |
|
157 |
if model == 'wizardlm':
|
158 |
model_repo = 'TheBloke/wizardLM-7B-HF'
|