Update app/main.py

app/main.py

@@ -1,37 +1,21 @@
 from fastapi import FastAPI, UploadFile, File, Request
-from fastapi.responses import HTMLResponse, JSONResponse
+from fastapi.responses import HTMLResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
 
 from app.infer import predict
 
-app = FastAPI(title="OCR Captcha Recognizer")
+app = FastAPI(title="Captcha OCR")
 
 app.mount("/static", StaticFiles(directory="static"), name="static")
 templates = Jinja2Templates(directory="templates")
 
-# -------------------------
-# HEALTH CHECK (HF NEEDS THIS)
-# -------------------------
-@app.get("/health")
-async def health():
-    return {"status": "ok"}
-
-# -------------------------
-# UI
-# -------------------------
 @app.get("/", response_class=HTMLResponse)
 async def home(request: Request):
-    return templates.TemplateResponse(
-        "index.html",
-        {"request": request}
-    )
+    return templates.TemplateResponse("index.html", {"request": request})
 
-# -------------------------
-# OCR API
-# -------------------------
 @app.post("/predict")
-async def predict_captcha(file: UploadFile = File(...)):
+async def predict_api(file: UploadFile = File(...)):
     image_bytes = await file.read()
     text = predict(image_bytes)
     return {"text": text}