Update app.py

app.py

@@ -13,9 +13,10 @@ 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}")
-        image = model(prompt=prompt, num_inference_steps=1, guidance_scale=0.0).images[0]
+        output = model(prompt=prompt, num_inference_steps=1, guidance_scale=0.0)
+        print(f"Model output: {output}")
+        image = output.images[0]
         buffered = BytesIO()
-        image.save(buffered, format="JPEG")
         image_bytes = buffered.getvalue()
         return image_bytes
     except Exception as e:
@@ -24,7 +25,7 @@ def generate_image(text, sentence_mapping, character_dict, selected_style):
 
 def inference(sentence_mapping, character_dict, selected_style):
     images = {}
-    print(f'sentence_mapping:{sentence_mapping}, character_dict:{character_dict}, selected_style:{selected_style}')
+    print(f'sentence_mapping: {sentence_mapping}, character_dict: {character_dict}, selected_style: {selected_style}')
     # Here we assume `sentence_mapping` is a dictionary where keys are paragraph numbers and values are lists of sentences
     grouped_sentences = sentence_mapping