Update app.py

app.py

@@ -11,12 +11,7 @@ model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
 
 # Helper function to truncate prompt to fit the model's maximum sequence length
 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
+    return prompt[:max_length]
 
 def generate_image(text, sentence_mapping, character_dict, selected_style):
     try:
@@ -28,6 +23,7 @@ def generate_image(text, sentence_mapping, character_dict, selected_style):
         output = model(prompt=prompt, num_inference_steps=1, guidance_scale=0.0)
         print(f"Model output: {output}")
         print("len of output:", len(output))
+        print("output.images:", output.images)
         # Check if the model returned images
         if output.images:
             image = output.images[0]