Update app.py

app.py

@@ -4,18 +4,11 @@ from utils import *
 from datasets import load_dataset
 import json
 
-pipe = pipeline(model="raminass/SCOTUS_AI_14", top_k=14, padding=True, truncation=True)
-all = load_dataset("raminass/full_opinions_1994_2020")
-df = pd.DataFrame(all["train"])
+pipe = pipeline(model="raminass/british", top_k=2, padding=True, truncation=True)
+df = pd.read_csv('data.csv', sep='\t')
 choices = []
-for index, row in df[df.category == "per_curiam"].iterrows():
-    if len(row["text"]) > 1000:
-        choices.append((f"""{row["case_name"]}""", [row["text"], row["year_filed"]]))
-
-with open("j_year.json", "r") as j:
-    judges_by_year = json.loads(j.read())
-judges_by_year = {int(k): v for k, v in judges_by_year.items()}
-
+for index, row in df.iterrows():
+    choices.append((f"""{row["case"]}""", [row["text"], row["author"]]))
 
 # https://www.gradio.app/guides/controlling-layout
 def greet(opinion, judges_l):