Solarum Asteridion commited on
Commit
6f9b73a
·
verified ·
1 Parent(s): d15f82f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ class LocalLLMHandler:
34
  self.tokenizer = None
35
  self.memory_tracker = MemoryTracker()
36
 
37
- def load_model(self, model_name="akjindal53244/Llama-3.1-Storm-8B"):
38
  try:
39
  if not setup_huggingface_auth():
40
  raise Exception("Hugging Face authentication failed. Please set your token first.")
 
34
  self.tokenizer = None
35
  self.memory_tracker = MemoryTracker()
36
 
37
+ def load_model(self, model_name="Qwen/Qwen2.5-1.5B-Instruct"):
38
  try:
39
  if not setup_huggingface_auth():
40
  raise Exception("Hugging Face authentication failed. Please set your token first.")