initial commit

app.py

@@ -12,9 +12,9 @@ from classy.utils.streamlit import get_md_200_random_color_generator
 
 
 def main(
-        model_checkpoint_path: str,
-        default_inventory_path: str,
-        cuda_device: int,
+    model_checkpoint_path: str,
+    default_inventory_path: str,
+    cuda_device: int,
 ):
     # setup examples
     examples = [
@@ -187,4 +187,8 @@ def main(
 
 
 if __name__ == "__main__":
-    main("experiments/extend-longformer-large/2021-10-22/09-11-39/checkpoints/best.ckpt", "data/inventories/aida.tsv", cuda_device=-1)
+    main(
+        "experiments/extend-longformer-large/2021-10-22/09-11-39/checkpoints/best.ckpt",
+        "data/inventories/aida.tsv",
+        cuda_device=-1,
+    )