Move demo files into subdirectory

README.md

@@ -5,7 +5,7 @@ colorFrom: gray
 colorTo: gray
 sdk: gradio
 sdk_version: 4.27.0
-app_file: app.py
+app_file: demo/app.py
 pinned: false
 language: en
 tags:
@@ -13,5 +13,4 @@ tags:
 - MARC
 - pytorch
 library_name: pytorch
-
 ---

app.py → demo/app.py

@@ -5,6 +5,9 @@ from marcai.utils.parsing import record_dict
 import pandas as pd
 from marcai.predict import predict_onnx
 from marcai.utils import load_config
+import os
+
+demo_dir = os.path.dirname(os.path.realpath(__file__))
 
 def compare(file1, file2):
     record1 = pymarc.parse_xml_to_array(file1)[0]
@@ -16,8 +19,8 @@ def compare(file1, file2):
     df = process(df1, df2)
 
     # Load model config
-    config = load_config("config.yaml")
-    model_onnx = "model.onnx"
+    config = load_config(os.path.join(demo_dir, "config.yaml"))
+    model_onnx = os.path.join(demo_dir, "model.onnx")
 
     # Run ONNX model
     input_df = df[config["model"]["features"]]
@@ -39,4 +42,4 @@ interface = gr.Interface(
     description="Upload two MARC XML files with one record each.",
     allow_flagging="never"
 )
-interface.launch()
+interface.launch()

setup.cfg

@@ -0,0 +1,14 @@
+[metadata]
+name = marcai
+version = 1.0.0
+
+[options]
+packages = find:
+
+[options.entry_points]
+console_scripts =
+    process = marcai:process.main
+    predict = marcai:predict.main
+    train = marcai:train.main
+    compare_records = marcai:compare_records.main
+