Update app.py

app.py

@@ -14,18 +14,20 @@ def truncate_prompt(prompt, max_length=77):
     tokens = prompt.split()
     if len(tokens) > max_length:
         return ' '.join(tokens[:max_length])
+    print("len of tokens:", len(tokens))
+    print("len of tokens:", len(prompt))
     return prompt
 
 def generate_image(text, sentence_mapping, character_dict, selected_style):
     try:
         prompt, _ = generate_prompt(text, sentence_mapping, character_dict, selected_style)
         print(f"Generated prompt: {prompt}")
-        
         # Truncate prompt if necessary
         prompt = truncate_prompt(prompt)
+        print(f"truncate_prompt: {prompt}")
         output = model(prompt=prompt, num_inference_steps=1, guidance_scale=0.0)
         print(f"Model output: {output}")
-
+        print("len of output:", len(output))
         # Check if the model returned images
         if output.images:
             image = output.images[0]