update

app.py

@@ -1,5 +1,6 @@
 from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel, Field
 from typing import Optional
 from cold.classifier import ToxicTextClassifier
@@ -21,6 +22,8 @@ class PredictionInput(BaseModel):
     text: str = Field(..., title="Text to classify", description="The text to classify for malicious content")
     context: Optional[str] = Field(None, title="Context for classification", description="Optional context to provide additional information for classification")
 
+app.mount("/", StaticFiles(directory="/app/out", html=True), name="static")
+
 @app.post("/predict")
 def predict(input: PredictionInput):
     try: