Update app.py

app.py

@@ -1,10 +1,11 @@
-from fastapi import FastAPI
-from datasets import load_dataset
-
-app = FastAPI()
-
-@app.get("/")
-def greet_json():
-    # Example: Loading a dataset as part of the API
-    dataset = load_dataset("squad")
-    return {"Hello": "World!", "dataset_length": len(dataset["train"])}
+from fastapi import FastAPI
+from datasets import load_dataset
+from transformers import AutoTokenizer
+
+app = FastAPI()
+
+@app.get("/")
+def greet_json():
+    # Example: Loading a dataset as part of the API
+    dataset = load_dataset("squad")
+    return {"Hello": "World!", "dataset_length": len(dataset["train"])}