Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ from utils import *
|
|
| 4 |
from datasets import load_dataset
|
| 5 |
import json
|
| 6 |
|
| 7 |
-
pipe = pipeline(model="raminass/
|
| 8 |
all = load_dataset("raminass/full_opinions_1994_2020")
|
| 9 |
df = pd.DataFrame(all["train"])
|
| 10 |
choices = []
|
|
|
|
| 4 |
from datasets import load_dataset
|
| 5 |
import json
|
| 6 |
|
| 7 |
+
pipe = pipeline(model="raminass/SCOTUS_AI_v2", top_k=14, padding=True, truncation=True)
|
| 8 |
all = load_dataset("raminass/full_opinions_1994_2020")
|
| 9 |
df = pd.DataFrame(all["train"])
|
| 10 |
choices = []
|