Update main.py
Browse files
main.py
CHANGED
|
@@ -15,11 +15,6 @@ nltk.download('punkt')
|
|
| 15 |
tokenizer = nltk.data.load('tokenizers/punkt/english.pickle')
|
| 16 |
|
| 17 |
HF_TOKEN = getenv("HF_TOKEN")
|
| 18 |
-
MODEL = "google/gemma-1.1-7b-it"
|
| 19 |
-
FALLBACK_MODELS = [
|
| 20 |
-
"mistralai/Mixtral-8x7B-Instruct-v0.1",
|
| 21 |
-
"mistralai/Mistral-7B-Instruct-v0.2", "mistralai/Mistral-7B-Instruct-v0.1"
|
| 22 |
-
]
|
| 23 |
|
| 24 |
class InputData(BaseModel):
|
| 25 |
model: str
|
|
|
|
| 15 |
tokenizer = nltk.data.load('tokenizers/punkt/english.pickle')
|
| 16 |
|
| 17 |
HF_TOKEN = getenv("HF_TOKEN")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
|
| 19 |
class InputData(BaseModel):
|
| 20 |
model: str
|