Update main.py

main.py

@@ -17,6 +17,7 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
+import asyncio
 import base64
 import json
 import logging
@@ -495,11 +496,14 @@ async def bing_dalle(item: BingDalle):
         except BaseException as e:
             return {"status": "false", "message": f"Internal Error: {e}"}
 
-    try:
-        response = await generate_image(item.prompt, item.model)
-        return response
-    except BaseException as e:
-        return {"status": "false", "message": f"Something went wrong: {e}"}
+    async def main():
+        try:
+            response = await generate_image(item.prompt, item.model)
+            return response
+        except BaseException as e:
+            return {"status": "false", "message": f"Something went wrong: {e}"}
+
+    return await main()
 
 
 @app.post("/UFoP/dalle3xl")