selectmixer
/

DepthFlowTest

Model card Files Files and versions Community

selectmixer commited on Aug 27, 2024

Commit

327c44c

·

1 Parent(s): 09e5be6

headless

Files changed (1) hide show

handler.py +1 -3

handler.py CHANGED Viewed

@@ -14,10 +14,8 @@ class EndpointHandler():
         Initialize the handler and load necessary resources.
         This method is called once when the service starts.
         """
-        os.environ['DISPLAY'] = ':0'  # Sometimes needed to trick OpenGL to run headless
         # Initialize DepthFlow once for efficiency
-        self.depthflow = DepthScene()
     def preprocess(self, inputs):
         """

         Initialize the handler and load necessary resources.
         This method is called once when the service starts.
         """
         # Initialize DepthFlow once for efficiency
+        self.depthflow = DepthScene(backend='headless')
     def preprocess(self, inputs):
         """