Root endpoint login

app.py

@@ -6,12 +6,17 @@ from pydantic import BaseModel
 from torch import cuda
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
+from huggingface_hub import login
+
 import logging
 # Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Required for access to a gated model
+login()
 
+# Keep data in session
 model = None
 tokenizer = None
 
@@ -129,3 +134,8 @@ async def health_check():
         "model_loaded": model is not None,
         "gpu_available": cuda.is_available()
     }
+
+@app.get("/")
+def read_root():
+    return PlainTextResponse('Habemus Apertus')
+