Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import AutoTokenizer,AutoModelForCausalLM,BitsAndBytesConfig
 import torch
-model_id = "rinna/japanese-gpt2-medium"
+model_id = "rinna/japanese-gpt2-small"
 tokenizer = AutoTokenizer.from_pretrained(model_id,trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(
     model_id,
@@ -25,7 +25,7 @@ def haiku_generate():
             pad_token_id=tokenizer.pad_token_id,
             do_sample=True,
             repetition_penalty=1.2,
-            temperature=1.0,
+            temperature=0.9,
             top_p=0.9
         )
     generated_tokens = outputs[0][prompt_len:]