kouki321 commited on
Commit
ea6d2c1
·
verified ·
1 Parent(s): 3ee5fd8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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 = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
 
96
  #"facebook/opt-125m"
97
  # Load tokenizer and model from disk (without trust_remote_code)
98
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
92
 
93
  # Initialize model and tokenizer
94
  def load_model_and_tokenizer():
95
+ model_name = "tiiuae/falcon-rw-1b"
96
+ #"deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
97
  #"facebook/opt-125m"
98
  # Load tokenizer and model from disk (without trust_remote_code)
99
  tokenizer = AutoTokenizer.from_pretrained(model_name)