Update app.py

app.py

@@ -2,10 +2,9 @@ import gradio as gr
 import cv2
 from geti_sdk.deployment import Deployment
 from geti_sdk.utils import show_image_with_annotation_scene
-import numpy as np
 
-# Step 1: Load the deployment
-deployment = Deployment.from_folder("deployment2")
+#Load models
+deployment = Deployment.from_folder("deployments")
 deployment.load_inference_models(device="CPU")
 
 def resize_image(image, target_dimension):
@@ -22,13 +21,13 @@ def infer(image):
         return None, 'Error: No image provided'
     
     image = resize_image(image, 1200)
-    image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
-    prediction = deployment.infer(image)
+    image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+    prediction = deployment.infer(image_rgb)
     output = show_image_with_annotation_scene(image, prediction, show_results=False)
     output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
     return output, prediction.overview
 
-interface = gr.Interface(
+demo = gr.Interface(
     fn=infer, 
     inputs="image",
     outputs=["image", "text"], 
@@ -37,4 +36,4 @@ interface = gr.Interface(
     examples=[["eggsample1.jpg"], ["eggsample2.jpg"]]
 )
 
-interface.launch(share=True)
+demo.launch()