felixz commited on
Commit
6120c47
·
1 Parent(s): fc74164

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import gradio as gr
2
  from transformers import T5Tokenizer, T5ForConditionalGeneration
3
 
4
- tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-large")
5
- model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-large")
6
 
7
  input_text = "translate English to German: How old are you?"
8
 
 
1
  import gradio as gr
2
  from transformers import T5Tokenizer, T5ForConditionalGeneration
3
 
4
+ tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-base")
5
+ model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-base")
6
 
7
  input_text = "translate English to German: How old are you?"
8