Spaces:
Sleeping
Sleeping
Solarum Asteridion
commited on
Update app.py
Browse files
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="
|
| 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.")
|