remove opencv dependencies

app.py

@@ -8,9 +8,7 @@
 @Contact :   2909171338@qq.com
 """
 
-import os
 import gradio as gr
-import cv2
 from PIL import Image
 import io
 import matplotlib.pyplot as plt
@@ -32,7 +30,6 @@ def get_dataset():
     """
     global ds
     if ds is None:
-        # ds = load_dataset("parquet", data_files={"train": "train/metadata.parquet", "test": "test/metadata.parquet"})
         ds = load_dataset("lh9171338/Wireframe")
     return ds
 
@@ -74,7 +71,7 @@ def draw_lines(image, lines):
     fig.add_axes(ax)
     plt.xlim([-0.5, width - 0.5])
     plt.ylim([height - 0.5, -0.5])
-    plt.imshow(image[:, :, ::-1])
+    plt.imshow(image)
     for pts in lines:
         pts = pts - 0.5
         plt.plot(pts[:, 0], pts[:, 1], color="orange", linewidth=0.5)
@@ -101,8 +98,7 @@ def show_image(split, index):
     """
     ds = get_dataset()
     example = ds[split][index]
-    image_file = os.path.join(split, example["file_name"])
-    image = cv2.imread(image_file)
+    image = np.array(example["image"])
     lines = np.array(example["lines"])
     image = draw_lines(image, lines)
     return image

requirements.txt

@@ -1,5 +1,4 @@
 datasets
 matplotlib
 numpy
-opencv-python
 pillow