Update app.py

app.py

@@ -62,11 +62,13 @@ def generate(
         progress=gr.Progress(track_tqdm=True),
 ):
     
-    input_ids = get_input_ids(LLM_PROMPT.replace("$USER_PROMPT", json.dumps(prompt)), BOT_PREFIX)
+    input_ids = get_input_ids(LLM_PROMPT.replace("$USER_PROMPT", json.dumps(prompt, ensure_ascii=False)), BOT_PREFIX)
     generation_kwargs = dict(input_ids=torch.tensor([input_ids]).to(llm.device), do_sample=True,
                              max_new_tokens=512, temperature=0.5, top_p=0.85, top_k=50, repetition_penalty=1.05)
     llm_result = llm.generate(**generation_kwargs)
-    print(tokenizer.decode(llm_result.cpu()[0], skip_special_tokens=True))
+    llm_result = BOT_PREFIX + tokenizer.decode(llm_result.cpu()[0], skip_special_tokens=True)
+    print(llm_result)
+    print(json.loads(llm_result))
 
     seed = random.randint(0, 2147483647)
     diffusion_pipe.to(device)