Upload folder using huggingface_hub

README.md

@@ -11,3 +11,11 @@ license: apache-2.0
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+## Run locally
+
+```bash
+cd demo
+pip install -r requirements.txt
+python3 app.py
+```

app.py

@@ -6,11 +6,12 @@ import gradio as gr
 import matplotlib.pyplot as plt
 import numpy as np
 from matplotlib.figure import Figure
+from PIL import Image
+
 from onnxtr.io import DocumentFile
 from onnxtr.models import ocr_predictor
 from onnxtr.models.predictor import OCRPredictor
 from onnxtr.utils.visualization import visualize_page
-from PIL import Image
 
 DET_ARCHS: List[str] = [
     "fast_base",
@@ -103,7 +104,7 @@ def forward_image(predictor: OCRPredictor, image: np.ndarray) -> np.ndarray:
 
 
 def matplotlib_to_pil(fig: Union[Figure, np.ndarray]) -> Image.Image:
-    """ Convert a matplotlib figure to a PIL image
+    """Convert a matplotlib figure to a PIL image
 
     Args:
     ----
@@ -136,7 +137,7 @@ def analyze_page(
     bin_thresh: float,
     box_thresh: float,
 ):
-    """ Analyze a page
+    """Analyze a page
 
     Args:
     ----
@@ -267,4 +268,4 @@ with gr.Blocks(fill_height=True) as demo:
         outputs=[input_image, segmentation_heatmap, output_image, ocr_output],
     )
 
-demo.launch(inbrowser=True, allowed_paths=['./data/logo.jpg'])
+demo.launch(inbrowser=True, allowed_paths=["./data/logo.jpg"])

requirements.txt

@@ -1,2 +1,2 @@
 -e git+https://github.com/felixdittrich92/OnnxTR.git#egg=onnxtr[cpu-headless,viz]
-gradio>=4.37.1,<5.0.0
+gradio>=4.37.1,<6.0.0