Update app.py

app.py

@@ -1,7 +1,10 @@
 import gradio as gr
 from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
-generator = pipeline('text-generation', model='https://gitee.com/dreamlikecloud/gpt2.git')
+tokenizer = AutoTokenizer.from_pretrained(pretrained_model_name_or_path="/root/.cache/huggingface/hub//models")
+model = AutoModelForCausalLM.from_pretrained(pretrained_model_name_or_path="/root/.cache/huggingface/hub//models")
+generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
 
 def generate(text):
     result = generator(text, max_length=30, num_return_sequences=1)