hertogateis commited on
Commit
cc86e76
·
verified ·
1 Parent(s): 6cc8274

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,6 +1,8 @@
1
  import streamlit as st
2
  import transformers
3
- from transformers import T5Tokenizer, T5ForConditionalGeneration
 
 
4
  #model_name = "t5-base"
5
  #tokenizer = T5Tokenizer.from_pretrained(model_name)
6
  #model = T5ForConditionalGeneration.from_pretrained(model_name)
 
1
  import streamlit as st
2
  import transformers
3
+ from transformers import GPT2LMHeadModel, GPT2Tokenizer
4
+
5
+ #from transformers import T5Tokenizer, T5ForConditionalGeneration
6
  #model_name = "t5-base"
7
  #tokenizer = T5Tokenizer.from_pretrained(model_name)
8
  #model = T5ForConditionalGeneration.from_pretrained(model_name)