Update app.py

app.py

@@ -15,12 +15,12 @@ repo = "stabilityai/stable-diffusion-3.5-large"
 t2i = StableDiffusion3Pipeline.from_pretrained(repo, torch_dtype=torch.bfloat16, token=os.environ["TOKEN"]).to(device)
 
 model = AutoModelForCausalLM.from_pretrained(
-    "llm-jp/llm-jp-3-1.8b-instruct", 
+    "google/gemma-2-2b-jpn-it", 
     device_map="cuda", 
     torch_dtype=torch.bfloat16, 
     trust_remote_code=True, 
 )
-tokenizer = AutoTokenizer.from_pretrained("llm-jp/llm-jp-3-1.8b-instruct")
+tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-2b-jpn-it")
 upsampler = pipeline(
     "text-generation",
     model=model,