Update app.py

app.py

@@ -1,24 +1,18 @@
 import gradio as gr
 import transformers
-from transformers import AutoModelWithLMHead, AutoTokenizer, pipeline
+from transformers import AutoModelForCausalLM, AutoModelWithLMHead, AutoTokenizer, pipeline
 from transformers import GPT2Tokenizer, GPT2Model
 
-model = GPT2Model.from_pretrained('gpt2')
-tokenizer = GPT2Tokenizer.from_pretrained(
-    'gpt2',
-)
 
-generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pretrained('gpt2'),
-    tokenizer='gpt2'#, 
-    #do_sample=True,
-    #max_length=120,
-    #top_k=50,
-    #top_p=0.95,
-    )
+general_model = AutoModelForCausalLM.from_pretrained('gpt2')
+general_generator = pipeline("text-generation", model=general_model, tokenizer="gpt2")
+general_result = general_generator("Today is ", max_length=120)
+general_result[0]["generated_text"]
+
 
 def generator(start_your_text = ''):
-  shroom_result = generator(start_your_text) 
-  return shroom_result[0]["generated_text"]
+  result = general_generator(start_your_text) 
+  return result[0]["generated_text"]
 
 iface = gr.Interface(fn=generator, inputs="text", outputs="text")
 iface.launch()