reset app.py

app.py

@@ -1,63 +1,9 @@
 import evaluate
-import json
-import os
-import re
-import sys
-from pathlib import Path
-
-module = evaluate.load("SEA-AI/horizon-metrics")
-
-REGEX_YAML_BLOCK = re.compile(r"---[\n\r]+([\S\s]*?)[\n\r]+---[\n\r]")
-
-
-def json_to_string_type(input_types):
-    """Maps json input type to str."""
-    return ["str" if i == "json" else i for i in input_types]
-
-
-def parse_readme(filepath):
-    """Parses a repositories README and removes"""
-    if not os.path.exists(filepath):
-        return "No README.md found."
-    with open(filepath, "r") as f:
-        text = f.read()
-        match = REGEX_YAML_BLOCK.search(text)
-        if match:
-            text = text[match.end():]
-    return text
-
-
-def launch_gradio_widget(metric):
-    """Launches `metric` widget with Gradio."""
-
-    try:
-        import gradio as gr
-    except ImportError as error:
-        print(
-            "To create a metric widget with Gradio make sure gradio is installed."
-        )
-        raise error
-
-    local_path = Path(sys.path[0])
-
-    def compute(data):
-        print(data)
-
-        metric.add(predictions=data, references=data)
-        return metric.compute(
-        )  ##**parse_gradio_data(data, gradio_input_types)
-
-    iface = gr.Interface(
-        description=
-        (metric.info.description +
-         "\nPass the predictions and the groud truth in the format specified below."
-         ),
-        title=f"Metric: {metric.name}",
-        article=parse_readme(local_path / "README.md"),
-        # examples=[parse_test_cases(test_cases, feature_names, gradio_input_types)]
-    )
-
-    iface.launch()
-
+from evaluate.utils import launch_gradio_widget
 
+module = evaluate.load("SEA-AI/horizon-metrics",
+                       roll_threshold=0.5,
+                       pitch_threshold=0.1,
+                       vertical_fov_degrees=25.6,
+                       height=512)
 launch_gradio_widget(module)