Update app.py

app.py

@@ -7,9 +7,9 @@ Description:
 FilePath: /zuo/app.py
 '''
 import gradio as gr
-from transformers import BertTokenizer, GPT2LMHeadModel,TextGenerationPipeline
-tokenizer = BertTokenizer.from_pretrained("./")
-model = GPT2LMHeadModel.from_pretrained("./")
+from transformers import BertTokenizer, TFGPT2LMHeadModel,TextGenerationPipeline
+tokenizer = BertTokenizer.from_pretrained("./tokenizer_config.json")
+model = TFGPT2LMHeadModel.from_pretrained("./config.json")
 text_generator = TextGenerationPipeline(model, tokenizer) 
 
 def poem(cls):