rcook commited on
Commit
9922904
·
verified ·
1 Parent(s): 4083964

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -40,6 +40,7 @@ from transformers import AutoTokenizer
40
 
41
  app = FastAPI()
42
 
 
43
  # Load dataset and tokenizer
44
  billsum = load_dataset("billsum", split="ca_test") # Load a small sample
45
  tokenizer = AutoTokenizer.from_pretrained("t5-small")
@@ -54,7 +55,7 @@ def preprocess_function(examples):
54
 
55
  return model_inputs
56
 
57
- @app.get("/")
58
  def get_tokenized_data():
59
  tokenized_billsum = billsum.map(preprocess_function, batched=True)
60
 
 
40
 
41
  app = FastAPI()
42
 
43
+ @app.get("/")
44
  # Load dataset and tokenizer
45
  billsum = load_dataset("billsum", split="ca_test") # Load a small sample
46
  tokenizer = AutoTokenizer.from_pretrained("t5-small")
 
55
 
56
  return model_inputs
57
 
58
+ #@app.get("/")
59
  def get_tokenized_data():
60
  tokenized_billsum = billsum.map(preprocess_function, batched=True)
61