DakMak commited on
Commit
3be58b8
1 Parent(s): 28cc1a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,8 +9,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
9
  suppress_st_warning=True, allow_output_mutation=True)
10
 
11
  def load_data():
12
- tokenizer = AutoTokenizer.from_pretrained("google/mt5-base")
13
- model = AutoModelForCausalLM.from_pretrained("google/mt5-base")
14
  return tokenizer, model
15
 
16
  tokenizer, model = load_data()
 
9
  suppress_st_warning=True, allow_output_mutation=True)
10
 
11
  def load_data():
12
+ tokenizer = AutoTokenizer.from_pretrained("Rubiksman1006/gpt-neo-2.7b-monika-fp16")
13
+ model = AutoModelForCausalLM.from_pretrained("Rubiksman1006/gpt-neo-2.7b-monika-fp16")
14
  return tokenizer, model
15
 
16
  tokenizer, model = load_data()