Update app.py

app.py

@@ -12,8 +12,7 @@ class Prodia:
     def __init__(self, api_key, base=None):
         self.base = base or "https://api.prodia.com/v2"
         self.headers = {
-            "X-Prodia-Key": api_key,
-            "Authorization": f"Bearer {os.getenv('API_KEY')}"
+            "X-Prodia-Key": api_key
         }
 
     def generate(self, params):
@@ -43,15 +42,18 @@ class Prodia:
 
 
     def generate_v2(self, config):
-        response = self._post("https://inference.prodia.com/v2/job", {"type": "v2.job.sdxl.txt2img", "config": config})
+        response = self._post("https://inference.prodia.com/v2/job", {"type": "v2.job.sdxl.txt2img", "config": config}, v2=True)
         return Image.open(BytesIO(r.content)).convert("RGBA")
         
 
-    def _post(self, url, params):
+    def _post(self, url, params, v2=False):
         headers = {
             **self.headers,
             "Content-Type": "application/json"
         }
+        if v2:
+            headers['Authorization'] = f"Bearer {os.getenv('API_KEY')}"
+            
         response = requests.post(url, headers=headers, data=json.dumps(params))
 
         if response.status_code != 200: