Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
| 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 |
|