Update app.py
Browse files
app.py
CHANGED
|
@@ -108,6 +108,21 @@ nltk.download('punkt')
|
|
| 108 |
|
| 109 |
FILES_DIR = './files'
|
| 110 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 111 |
# Model Management
|
| 112 |
class ModelManager:
|
| 113 |
def __init__(self):
|
|
|
|
| 108 |
|
| 109 |
FILES_DIR = './files'
|
| 110 |
|
| 111 |
+
|
| 112 |
+
DEFAULT_MODELS = {
|
| 113 |
+
"HuggingFace": [
|
| 114 |
+
"paraphrase-miniLM",
|
| 115 |
+
"paraphrase-mpnet",
|
| 116 |
+
"all-MiniLM-L6-v2"
|
| 117 |
+
],
|
| 118 |
+
"OpenAI": [
|
| 119 |
+
"text-embedding-ada-002"
|
| 120 |
+
],
|
| 121 |
+
"Cohere": [
|
| 122 |
+
"embed-multilingual-v2.0"
|
| 123 |
+
]
|
| 124 |
+
}
|
| 125 |
+
|
| 126 |
# Model Management
|
| 127 |
class ModelManager:
|
| 128 |
def __init__(self):
|