Update app.py

app.py

@@ -2,6 +2,8 @@ from fastapi import FastAPI
 from pydantic import BaseModel
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
+from fastapi.staticfiles import StaticFiles
+
 
 # -------------------------------
 # Load model & tokenizer from HF Hub
@@ -16,6 +18,8 @@ device = "cpu"  # Spaces free tier uses CPU; you can switch to "cuda" if GPU gra
 # FastAPI setup
 # -------------------------------
 app = FastAPI()
+app.mount("/", StaticFiles(directory="static", html=True), name="static")
+
 
 class Query(BaseModel):
     text: str