Update app.py

app.py

@@ -4,21 +4,19 @@ import cv2
 import requests
 import json
 import gradio as gr
-
-# Define a function to call the API and get the results
 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.")
+# Checking if the environment variable exists
+if not api_key:
+    print("PXiVision environment variable is not set.")
+    exit()
 
+# Define a function to call the API and get the results
 def get_results(image):
-    threshold=0.5
+    threshold = 0.5
     # Convert the image to base64 string
     with io.BytesIO() as output:
         image.save(output, format="JPEG")
@@ -47,7 +45,7 @@ def get_results(image):
 
 # Define the input and output components for Gradio
 image_input = gr.inputs.Image()  # Adjust the shape according to your requirements
-output_component = gr.outputs.Label(num_top_classes=len(output_labels))
+output_component = gr.outputs.Label()
 
 # Launch the Gradio interface
-gr.Interface(fn=get_results, inputs=image_input, outputs=output_component).launch()
+gr.Interface(fn=get_results, inputs=image_input, outputs=output_component).launch()