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(
-    "microsoft/Phi-3.5-mini-instruct", 
+    "llm-jp/llm-jp-3-1.8b-instruct", 
     device_map="cuda", 
     torch_dtype=torch.bfloat16, 
     trust_remote_code=True, 
 )
-tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3.5-mini-instruct")
+tokenizer = AutoTokenizer.from_pretrained("llm-jp/llm-jp-3-1.8b-instruct")
 upsampler = pipeline(
     "text-generation",
     model=model,
@@ -28,7 +28,7 @@ upsampler = pipeline(
 )
 
 generation_args = {
-    "max_new_tokens": 300,
+    "max_new_tokens": 70,
     "return_full_text": False,
     "temperature": 0.7,
     "do_sample": True,