Fix OMP_NUM_THREADS warning by setting it before torch import

app.py

@@ -5,7 +5,12 @@ Text-to-Image and Text-to-Video Generation
 This app allows you to run STARFlow inference on Hugging Face GPU infrastructure.
 """
 
+# Fix OpenMP warning - MUST be set BEFORE importing torch
 import os
+os.environ['OMP_NUM_THREADS'] = '1'
+os.environ['MKL_NUM_THREADS'] = '1'
+os.environ['NUMEXPR_NUM_THREADS'] = '1'
+
 import warnings
 import gradio as gr
 import torch
@@ -16,9 +21,6 @@ from pathlib import Path
 # Suppress harmless warnings
 warnings.filterwarnings("ignore", category=FutureWarning, message=".*torch.distributed.reduce_op.*")
 
-# Fix OpenMP warning
-os.environ['OMP_NUM_THREADS'] = '1'
-
 # Try to import huggingface_hub for downloading checkpoints
 try:
     from huggingface_hub import hf_hub_download