cleanup

app.py

@@ -8,7 +8,6 @@ from pathlib import Path
 from transformers import AutoProcessor, AutoModelForVision2Seq
 import subprocess
 import logging
-subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -403,10 +402,10 @@ def create_ui(examples_path: str, model_path: str):
     return app
 
 if __name__ == "__main__":
+    subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+
     # Initialize CUDA
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     
-    MODEL_PATH = os.getenv("MODEL_PATH", "HuggingFaceTB/SmolVLM2-2.2B-Instruct")
-    
-    app = create_ui("video_spec.json", MODEL_PATH)
+    app = create_ui("video_spec.json", "HuggingFaceTB/SmolVLM2-2.2B-Instruct")
     app.launch()