Update app.py

app.py

@@ -6,7 +6,7 @@ model_name = "rinna/japanese-gpt2-medium"
 
 # トークナイザーとパイプラインの設定
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-generator = pipeline('text-generation', model=model_name, tokenizer=tokenizer, device=-1)  # device=0はGPUを使用する設定
+generator = pipeline('text-generation', model=model_name, tokenizer=tokenizer, device=0)  # device=0はGPUを使用する設定
 
 def generate_text(prompt, max_length):
     result = generator(prompt, max_length=max_length, num_return_sequences=1)