loocorez commited on
Commit
7d7ef5b
·
verified ·
1 Parent(s): 96cfd49

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -11,9 +11,10 @@ import gradio as gr
11
  from nanochat.checkpoint_manager import load_model_from_dir
12
  from nanochat.engine import Engine
13
 
14
- MODEL_REPO = os.getenv("NANOCHAT_MODEL_REPO", "loocorez/nanochat-base-d20-step21400")
15
- STEP = os.getenv("NANOCHAT_STEP", "021400")
16
- DEPTH = os.getenv("NANOCHAT_DEPTH", "20")
 
17
 
18
  ckpt_dir = f"/tmp/ckpt/d{DEPTH}"
19
  os.makedirs(ckpt_dir, exist_ok=True)
 
11
  from nanochat.checkpoint_manager import load_model_from_dir
12
  from nanochat.engine import Engine
13
 
14
+ # Hardcoded model selection for this Space
15
+ MODEL_REPO = "loocorez/nanochat-base-d20-step21400"
16
+ STEP = "021400"
17
+ DEPTH = "20"
18
 
19
  ckpt_dir = f"/tmp/ckpt/d{DEPTH}"
20
  os.makedirs(ckpt_dir, exist_ok=True)