Update app.py

app.py

@@ -67,15 +67,16 @@ def generate(
                              max_new_tokens=512, temperature=0.5, top_p=0.85, top_k=50, repetition_penalty=1.05)
     llm_result = llm.generate(**generation_kwargs)
     llm_result = BOT_PREFIX + tokenizer.decode(llm_result.cpu()[0], skip_special_tokens=True)
+    expanded_description = json.loads(llm_result)["expanded_description"]
     print(llm_result)
-    print(json.loads(llm_result))
+    print(expanded_description)
 
     seed = random.randint(0, 2147483647)
     diffusion_pipe.to(device)
     generator = torch.Generator().manual_seed(seed)
 
     images = diffusion_pipe(
-        prompt=prompt,
+        prompt=expanded_description,
         negative_prompt=None,
         width=1024,
         height=1024,