MD1998 commited on
Commit
185f6c5
·
verified ·
1 Parent(s): 77f480c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -45,6 +45,8 @@ import gradio as gr
45
 
46
  ##########################
47
 
 
 
48
 
49
  finetuned_model = T5ForConditionalGeneration.from_pretrained("MD1998/chating_beginner_v2")
50
  tokenizer = T5Tokenizer.from_pretrained("MD1998/chating_beginner_v2")
 
45
 
46
  ##########################
47
 
48
+ from transformers import T5Tokenizer
49
+ from transformers import T5ForConditionalGeneration
50
 
51
  finetuned_model = T5ForConditionalGeneration.from_pretrained("MD1998/chating_beginner_v2")
52
  tokenizer = T5Tokenizer.from_pretrained("MD1998/chating_beginner_v2")