Merge pull request #1 from prajinkhadka/main

Add batch size in the Constructor.

Author: ezzcodeezzlife

src/dalle2/dalle2.py

@@ -6,8 +6,9 @@
 import os
 
 class Dalle2():
-    def __init__(self, bearer):
+    def __init__(self, bearer, batch_size):
         self.bearer = bearer
+        self.batch_size = batch_size
 
     def generate(self, promt):
         url = "https://labs.openai.com/api/labs/tasks"
@@ -19,7 +20,7 @@ def generate(self, promt):
             "task_type": "text2im",
             "prompt": {
                 "caption": promt,
-                "batch_size": 6,
+                "batch_size": self.batch_size,
             }
         }