Hugging Face's logo

Spaces:
patpizio
/
llm-token-probs
Sleeping

App Files Community
patpizio commited on
Commit
21c2259
·
1 Parent(s): 74fd993

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, T5Tokenizer, T5ForConditionalGeneration,
4
 
5
  st.title('How does the LLM choose its words?')
6
 
7
- model_checkpoint = "google/flan-t5-small"
8
 
9
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
10
 
 
4
 
5
  st.title('How does the LLM choose its words?')
6
 
7
+ model_checkpoint = "google/flan-t5-base"
8
 
9
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
10