update

Dockerfile

@@ -7,7 +7,9 @@ ARG HF_TOKEN
 ENV DEBIAN_FRONTEND=noninteractive \
     PYTHONUNBUFFERED=1 \
     PYTHONDONTWRITEBYTECODE=1 \
-    HF_TOKEN=${HF_TOKEN}
+    HF_TOKEN=${HF_TOKEN} \
+    PYOPENGL_PLATFORM=osmesa \
+    DISPLAY=:99
 
 WORKDIR /code
 
@@ -20,6 +22,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     libomp-dev \
     libosmesa6-dev \
     libglib2.0-0 \
+    libx11-6 \
+    libxext6 \
+    libxrender1 \
+    libgl1-mesa-dri \
     && rm -rf /var/lib/apt/lists/*
 
 # Copy requirements and install Python dependencies

app.py

@@ -1,3 +1,6 @@
+import os
+os.environ['PYOPENGL_PLATFORM'] = 'osmesa'
+
 from fastapi import FastAPI, HTTPException
 from fastapi.responses import StreamingResponse
 from pydantic import BaseModel, Field

renderer.py

@@ -1,3 +1,6 @@
+import os
+os.environ['PYOPENGL_PLATFORM'] = 'osmesa'
+
 import numpy as np
 import pyrender
 import trimesh
@@ -14,7 +17,11 @@ class AvatarRenderer:
         self.image_size = image_size
         self.camera_type = camera_type
         self.light_intensity = light_intensity
-        self.renderer = pyrender.OffscreenRenderer(image_size, image_size)
+        self.renderer = pyrender.OffscreenRenderer(
+            viewport_width=image_size,
+            viewport_height=image_size,
+            point_size=1.0
+        )
     
     def render(
         self,