Update app.py
Browse files
app.py
CHANGED
|
@@ -92,7 +92,8 @@ def clean_up(cache, origin_len):
|
|
| 92 |
|
| 93 |
# Initialize model and tokenizer
|
| 94 |
def load_model_and_tokenizer():
|
| 95 |
-
model_name = "
|
|
|
|
| 96 |
#"deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
|
| 97 |
#"facebook/opt-125m"
|
| 98 |
# Load tokenizer and model from disk (without trust_remote_code)
|
|
|
|
| 92 |
|
| 93 |
# Initialize model and tokenizer
|
| 94 |
def load_model_and_tokenizer():
|
| 95 |
+
model_name = "Locutusque/TinyMistral-248M"
|
| 96 |
+
#"tiiuae/falcon-rw-1b"
|
| 97 |
#"deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
|
| 98 |
#"facebook/opt-125m"
|
| 99 |
# Load tokenizer and model from disk (without trust_remote_code)
|