Update app.py

app.py

@@ -10,16 +10,16 @@ import json
 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):
-    return prompt[:max_length]
+# def truncate_prompt(prompt, max_length=77):
+#     return prompt[:max_length]
 
 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}")
+        # 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))
@@ -50,8 +50,8 @@ def inference(sentence_mapping, character_dict, selected_style):
             combined_sentence = " ".join(sentences)
             futures[paragraph_number] = executor.submit(generate_image, combined_sentence, sentence_mapping, character_dict, selected_style)
 
-        for paragraph_number, future in futures.items():
-            images[paragraph_number] = future.result()
+    for paragraph_number, future in futures.items():
+        images[paragraph_number] = future.result()
 
     return images