healthcheck endpoint

main.py

@@ -33,3 +33,12 @@ def translate(req: TranslationRequest):
     translated = model.generate(**inputs)
     result = tokenizer.decode(translated[0], skip_special_tokens=True)
     return {"translation": result}
+
+@app.get("/healthcheck")
+def healthcheck():
+    # Run a dummy translation to warm up the model
+    test_text = "Hello"
+    inputs = tokenizer([test_text], return_tensors="pt", padding=True, truncation=True)
+    translated = model.generate(**inputs)
+    result = tokenizer.decode(translated[0], skip_special_tokens=True)
+    return {"status": "ok", "test_translation": result}