Update app.py

app.py

@@ -1,10 +1,13 @@
 import gradio as gr
-from transformers import pipeline
-from transformers import AutoTokenizer, AutoModelForCausalLM
 
-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)
+# from transformers import pipeline
+# from transformers import AutoTokenizer, AutoModelForCausalLM
+
+# 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)
+
+generator = pipeline('text-generation', model='gpt2')
 
 def generate(text):
     result = generator(text, max_length=30, num_return_sequences=1)