Update app.py

app.py

@@ -1,7 +1,19 @@
 from fastapi import FastAPI
+from pydantic import BaseModel
+from transformers import pipeline
 
 app = FastAPI()
 
-@app.get("/")
-def greet_json():
-    return {"Hello": "World!"}
+# Load your model
+clf = pipeline(
+    "text-classification",
+    model="DelaliScratchwerk/time-period-classifier-bert"
+)
+
+class PredictRequest(BaseModel):
+    inputs: str
+
+@app.post("/predict")
+def predict(req: PredictRequest):
+    result = clf(req.inputs)
+    return result