Update app.py

app.py

@@ -6,10 +6,10 @@ import transformers
 from transformers import AutoModelWithLMHead, AutoTokenizer, AutoModelForCausalLM, pipeline
 from transformers import GPT2Tokenizer, GPT2Model
 
-tokenizer = AutoTokenizer.from_pretrained("samroni/model_gpt")
+tokenizer = AutoTokenizer.from_pretrained("samroni/puisi_model_gpt2_small")
 
-model = AutoModelForCausalLM.from_pretrained("samroni/model_gpt")
-pipe = pipeline('text-generation', model="samroni/model_gpt", tokenizer=tokenizer)
+model = AutoModelForCausalLM.from_pretrained("samroni/puisi_model_gpt2_small")
+pipe = pipeline('text-generation', model="samroni/puisi_model_gpt2_small", tokenizer=tokenizer)
 
 
 def text_generation(input_text, seed):
@@ -19,8 +19,8 @@ def text_generation(input_text, seed):
   generated_text = tokenizer.batch_decode(outputs, skip_special_tokens=True)
   return generated_text
 
-title = "Text Generator Demo GPT2"
-description = "Text Generator Application by ecarbo"
+title = "Indonesia Poem Generator Demo GPT2"
+description = "Poem Generator "
 
 gr.Interface(
     text_generation,