Static html v1.

app.py

@@ -1,14 +1,15 @@
 from fastapi import FastAPI, Request
 from fastapi.responses import HTMLResponse, StreamingResponse
 from fastapi.templating import Jinja2Templates
+from fastapi.staticfiles import StaticFiles
 from pmbl import PMBL
-
-
-
+import os
 
 app = FastAPI()
 pmbl = PMBL("./PMB-7b.Q6_K.gguf")  # Replace with the path to your model
 
+app.mount("/", StaticFiles(directory="."), name="static")
+
 templates = Jinja2Templates(directory=".")
 
 @app.post("/chat")
@@ -37,11 +38,9 @@ async def sleep():
         print(f"[SYSTEM] Error: {str(e)}")
         return {"error": str(e)}
 
-# Remove the main function as Hugging Face Spaces will automatically run the app. Using here to test
-
 if __name__ == "__main__":
     import uvicorn
     import asyncio
 
     loop = asyncio.get_event_loop()
-    loop.run_until_complete(uvicorn.run(app, host="0.0.0.0", port=1771))
+    loop.run_until_complete(uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", 8000))))