davidizzle commited on
Commit
1b9ef5d
·
1 Parent(s): 0c1dfa2

Changed to non-gated model for demo purposes

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -37,7 +37,9 @@ gif_html.markdown(
37
 
38
  @st.cache_resource
39
  def load_model():
40
- model_id = "google/gemma-2b-it"
 
 
41
  tokenizer = AutoTokenizer.from_pretrained(model_id, token=True)
42
  model = AutoModelForCausalLM.from_pretrained(
43
  model_id,
 
37
 
38
  @st.cache_resource
39
  def load_model():
40
+ # Access to Gemma is gated, so the Space will not inherit it. Select tiny gpt2 to stub
41
+ # model_id = "google/gemma-2b-it"
42
+ model_id = "sshleifer/tiny-gpt2" # OR distilgpt2
43
  tokenizer = AutoTokenizer.from_pretrained(model_id, token=True)
44
  model = AutoModelForCausalLM.from_pretrained(
45
  model_id,