Fix torch/torchao/torchvision versions for spaces.aoti_capture

- spaces.aoti_capture requires PyTorch 2.8+
- Install torch>=2.8,<2.9 with matching torchvision>=0.23,<0.24
- Install torchao at runtime (compatible with torch 2.8)

app.py

@@ -4,10 +4,11 @@ Gradio Demo for Chinese Calligraphy Generation - HuggingFace Space Version
 With Float8 quantization and AOT compilation for faster inference
 """
 
-# Install compatible torch + torchvision + spaces versions (following FLUX-Kontext-fp8 pattern)
+# Install compatible torch 2.8 + torchvision 0.23 + torchao + spaces (for AOT compilation)
+# spaces.aoti_capture requires PyTorch 2.8+
 import os
-os.system('pip install --upgrade --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu126 "torch<2.9" "torchvision<0.20" spaces')
-print("torch + torchvision + spaces installation complete!")
+os.system('pip install --upgrade --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu126 "torch>=2.8,<2.9" "torchvision>=0.23,<0.24" torchao spaces')
+print("torch 2.8 + torchvision 0.23 + torchao + spaces installation complete!")
 
 # IMPORTANT: import spaces first before any CUDA-related packages
 import spaces

requirements.txt

@@ -1,5 +1,5 @@
-# AOT compilation dependencies (same as FLUX-Kontext-fp8)
-torchao==0.11
+# AOT compilation dependencies
+# torchao is installed at runtime with torch 2.8 compatible version (see app.py)
 transformers
 git+https://github.com/huggingface/diffusers.git
 accelerate