Upload main.py

main.py

@@ -40,11 +40,14 @@ templates = Jinja2Templates(directory=str(BASE_DIR / "templates"))
 
 @app.get("/", response_class=HTMLResponse)
 async def read_root(request: Request):
-    return templates.TemplateResponse("index.html", {
-        "request": request, 
-        "model_loaded": current_model is not None, 
-        "model_name": model_name
-    })
+    return templates.TemplateResponse(
+        request=request,
+        name="index.html",
+        context={
+            "model_loaded": current_model is not None, 
+            "model_name": model_name
+        }
+    )
 
 @app.post("/upload-model")
 async def upload_model(file: UploadFile = File(...)):