Update app.py

app.py

@@ -6,7 +6,18 @@ import json
 import gradio as gr
 
 # Define a function to call the API and get the results
-def get_results(imag):
+import os
+
+# Accessing a specific environment variable
+api_key = os.environ.get('PXiVision')
+
+# Checking if an environment variable exists
+if 'PXiVision' in os.environ:
+    api_key = os.environ['PXiVision']
+else:
+    print("API_KEY environment variable is not set.")
+
+def get_results(image):
     threshold=0.5
     # Convert the image to base64 string
     with io.BytesIO() as output:
@@ -17,7 +28,7 @@ def get_results(imag):
     payload = json.dumps({"base64str": base64str, "threshold": threshold})
 
     # Send the request to the API
-    response = requests.put("", data=payload)
+    response = requests.put(api_key, data=payload)
 
     # Parse the JSON response
     data = response.json()