Update app.py

app.py

@@ -86,7 +86,7 @@ def get_result(text_prompts, style_indx):
     #img_nparray= cv2.cvtColor(img_np, cv2.COLOR_BGR2RGBA)
     #img_blue_correction = Image.fromarray(img_nparray)
     #img_blue_correction.save(imagefile, img_blue_correction.format)
-    image_output[0].save(imagefile, image_output[0].format)
+    image_output.save(imagefile, image_output.format)
     
     interrogate_prompt = img_to_text(imagefile, fn_index=1)[0]
     print(interrogate_prompt)