minor correction

App/Generate/database/Model.py

@@ -260,8 +260,11 @@ class Scene(orm.Model):
                 while True:
                     result = await image_generator.generate_image(payload)
                     status = await image_generator.fetch_image_status(result["id"])
-                    if status["output"]:
+
+                    if type(status["output"]) == list:
                         self.images.extend(status["output"])
+                    else:
+                        self.images.append(status["output"])
                         break