ghosthets commited on
Commit
620d411
·
verified ·
1 Parent(s): 7b10240

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ app = flask.Flask(__name__)
12
  # ===========================
13
  # LOAD MODEL (StableLM-3B-Chat)
14
  # ===========================
15
- model_id = "stabilityai/StableLM-3B-4E1T-v2"
16
  print("🔄 Loading model...")
17
 
18
  # CPU/GPU device set
 
12
  # ===========================
13
  # LOAD MODEL (StableLM-3B-Chat)
14
  # ===========================
15
+ model_id = "HuggingFaceTB/SmolLM-1.7B"
16
  print("🔄 Loading model...")
17
 
18
  # CPU/GPU device set