JustKiddo commited on
Commit
13846fb
·
verified ·
1 Parent(s): 038efdf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,8 +12,8 @@ client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
12
  dataset = load_dataset("JustKiddo/KiddosVault")
13
 
14
  # Load the tokenizer and model for token display
15
- tokenizer = AutoTokenizer.from_pretrained("t5-small") #Google's T5 Model
16
- model = AutoModelForSeq2SeqLM.from_pretrained("t5-small")
17
 
18
  def respond(
19
  message,
 
12
  dataset = load_dataset("JustKiddo/KiddosVault")
13
 
14
  # Load the tokenizer and model for token display
15
+ tokenizer = AutoTokenizer.from_pretrained("google-t5/t5-small") #Google's T5 Model
16
+ model = AutoModelForSeq2SeqLM.from_pretrained("google-t5/t5-small")
17
 
18
  def respond(
19
  message,