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)
|