Update app.py

app.py

@@ -3,9 +3,9 @@ from transformers import pipeline, AutoTokenizer
 import spaces
 @spaces.GPU
 # generator = pipeline('text-generation', model='distilgpt2', device=-1)
-model_name = "rinna/japanese-gpt2-medium"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-generator = pipeline('text-generation', model=model_name, tokenizer=tokenizer, device=-1)
+# model_name = "rinna/japanese-gpt2-medium"
+tokenizer = AutoTokenizer.from_pretrained('rinna/japanese-gpt2-medium')
+generator = pipeline('text-generation', 'rinna/japanese-gpt2-medium', tokenizer=tokenizer, device=-1)
 
 def generate_text(prompt, max_length):
     result = generator(prompt, max_length=max_length, num_return_sequences=1)