dfalbel commited on
Commit
93d628c
1 Parent(s): 4afd74e

Allow setting the model repo via env var.

Browse files
Files changed (1) hide show
  1. app.R +2 -2
app.R CHANGED
@@ -4,8 +4,8 @@ library(minhub)
4
  library(magrittr)
5
  source("model-session.R")
6
 
7
- repo <- "stabilityai/stablelm-tuned-alpha-3b"
8
- #repo <- "EleutherAI/pythia-70m"
9
  sess <- model_session$new()
10
 
11
  max_n_tokens <- 100
 
4
  library(magrittr)
5
  source("model-session.R")
6
 
7
+ repo <- "EleutherAI/pythia-70m"
8
+ repo <- Sys.getenv("MODEL_REPO", unset = repo)
9
  sess <- model_session$new()
10
 
11
  max_n_tokens <- 100