Update app.py

app.py

@@ -22,7 +22,7 @@ def generate_image(prompt):
         print(f"Generating image with truncated prompt: {truncated_prompt}")
         
         # Call the model
-        output = model(prompt=truncated_prompt, num_inference_steps=50, guidance_scale=7.5)
+        output = model(prompt=truncated_prompt, num_inference_steps=1, guidance_scale=0.0)
         
         # Check if output is valid
         if output is not None and hasattr(output, 'images') and output.images:
@@ -54,15 +54,10 @@ def process_prompt(sentence_mapping, character_dict, selected_style):
     print(f"Character Dict: {character_dict}")
     print(f"Selected Style: {selected_style}")
 
-    prompts = []
+    images = {}
     for paragraph_number, sentences in sentence_mapping.items():
         combined_sentence = " ".join(sentences)
-        prompt = generate_prompt(combined_sentence, sentence_mapping, character_dict, selected_style)
-        prompts.append((paragraph_number, prompt))
-        print(f"Generated prompt for paragraph {paragraph_number}: {prompt}")
-
-    images = {}
-    for paragraph_number, prompt in prompts:
+        prompt, _ = generate_prompt(combined_sentence, sentence_mapping, character_dict, selected_style)
         img_str = inference(prompt)
         images[paragraph_number] = img_str