Upload with huggingface_hub

app.py

@@ -0,0 +1,46 @@
+# Information extraction from a typed data specification.
+
+import minichain
+from dataclasses import dataclass
+from typing import List
+from enum import Enum
+
+# Data specification
+
+# +
+class StatType(Enum):
+    POINTS = 1
+    REBOUNDS = 2
+    ASSISTS = 3
+
+@dataclass
+class Stat:
+    value: int
+    stat: StatType
+
+@dataclass
+class Player:
+    player: str
+    stats: List[Stat]
+# -
+
+
+# Code
+
+class ExtractionPrompt(minichain.TypedTemplatePrompt):
+    template_file = "stats.pmpt.tpl"
+    Out = Player
+
+
+with minichain.start_chain("stats") as backend:
+    p = ExtractionPrompt(backend.OpenAI(max_tokens=512))
+    article = open("sixers.txt").read()
+    for player in p({"passage": article}):
+        print(player)
+
+ExtractionPrompt().show({"passage": "Harden had 10 rebounds."},
+                        '[{"player": "Harden", "stats": {"value": 10, "stat": 2}}]')
+
+# View the run log.
+
+minichain.show_log("bash.log")

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+git+https://github.com/srush/minichain
+manifest-ml

stats.pmpt.tpl

@@ -0,0 +1,6 @@
+{{typ | safe}}
+
+{{passage}}
+
+
+JSON Output: