fix a bug

app.py

@@ -1,4 +1,5 @@
 import cv2
+import paddle
 import paddlehub as hub
 import gradio as gr
 
@@ -7,19 +8,24 @@ module = hub.Module(name="lseg")
 
 
 def segment(image, labels):
-    long_size = max(image.shape[:2])
-    if long_size > 512:
-        f = 512 / long_size
-        image = cv2.resize(image, (0, 0), fx=f, fy=f)
-    results = module.segment(
-        image=image[..., ::-1],
-        labels=[item for item in labels.split('\n') if item != '']
-    )
-    return [
-        results['color'][..., ::-1],
-        results['mix'][..., ::-1],
-        *[v[..., ::-1] for v in results['classes'].values()]
-    ]
+    try:
+        long_size = max(image.shape[:2])
+        if long_size > 512:
+            f = 512 / long_size
+            image = cv2.resize(image, (0, 0), fx=f, fy=f)
+        results = module.segment(
+            image=image[..., ::-1],
+            labels=[item for item in labels.split('\n') if item != '']
+        )
+
+        return [
+            results['color'][..., ::-1],
+            results['mix'][..., ::-1],
+            *[cv2.cvtColor(v, cv2.COLOR_BGRA2RGBA) for v in results['classes'].values()]
+        ]
+    except:
+        paddle.disable_static()
+        raise ValueError
 
 
 gr.Interface(