Update app.py

app.py

@@ -83,9 +83,11 @@ def get_tryon_result(model_name, garment1, garment2, seed=1234):
 
     # Fix for the missing environment variable
 
+  
+    
 
     try:
-        url = os.environ.get("OA_IP_ADDRESS")  # reads your key from Hugging Face secrets
+        url = os.environ['OA_IP_ADDRESS']  # reads your key from Hugging Face secrets
     except KeyError:
         print("Error: OA_IP_ADDRESS environment variable not set!")
         # return a dummy image instead of crashing
@@ -93,10 +95,10 @@ def get_tryon_result(model_name, garment1, garment2, seed=1234):
         dummy_img = cv2.putText(dummy_img, 'Error: API key not set', (50, 256),
                                 cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 2)
         return add_waterprint(dummy_img)
-    
-    headers = {
-        'Content-Type': 'application/json'
-    }
+
+    headers = {'Content-Type': 'application/json',
+              "Authorization": f"Bearer {os.environ.get('OA_API_KEY', '')}"
+              }
 
     seed = random.randint(0, 1222222222)
     data = {