added versions to requirements and adjusted startup to be more efficient

app.py

@@ -28,30 +28,21 @@ model_3n_id = os.getenv("MODEL_3N_ID", "google/gemma-3n-E4B-it")
 MAX_VIDEO_SIZE = 100 * 1024 * 1024  # 100 MB
 MAX_IMAGE_SIZE = 10 * 1024 * 1024   # 10 MB
 
-# Skip model loading during tests
-SKIP_MODEL_LOADING = os.getenv("SKIP_MODEL_LOADING", "false").lower() == "true"
+input_processor = Gemma3Processor.from_pretrained(model_12_id)
 
-if not SKIP_MODEL_LOADING:
-    input_processor = Gemma3Processor.from_pretrained(model_12_id)
-
-    model_12 = Gemma3ForConditionalGeneration.from_pretrained(
-        model_12_id,
-        torch_dtype=torch.bfloat16,
-        device_map="auto",
-        attn_implementation="eager",
-    )
+model_12 = Gemma3ForConditionalGeneration.from_pretrained(
+    model_12_id,
+    torch_dtype=torch.bfloat16,
+    device_map="auto",
+    attn_implementation="eager",
+)
 
-    model_3n = Gemma3nForConditionalGeneration.from_pretrained(
-        model_3n_id,
-        torch_dtype=torch.bfloat16,
-        device_map="auto",
-        attn_implementation="eager",
-    )
-else:
-    # Mock objects for testing
-    input_processor = None
-    model_12 = None
-    model_3n = None
+model_3n = Gemma3nForConditionalGeneration.from_pretrained(
+    model_3n_id,
+    torch_dtype=torch.bfloat16,
+    device_map="auto",
+    attn_implementation="eager",
+)
 
 
 def check_file_size(file_path: str) -> bool:

requirements.txt

@@ -1,11 +1,11 @@
-torch
-spaces
-gradio
-transformers
-accelerate
-pytest
-loguru
-python-dotenv
-opencv-python
-timm
-pymupdf
+torch==2.6.0+cu126
+spaces==0.40.0
+gradio==5.42.0
+transformers==4.55.2
+accelerate==1.10.0
+pytest==8.4.1
+loguru==0.7.3
+python-dotenv==1.1.1
+opencv-python==4.12.0.88
+timm==1.0.19
+pymupdf==1.26.3

tests/conftest.py

@@ -1,12 +0,0 @@
-import os
-import pytest
-
-# Set environment variable before any imports
-os.environ["SKIP_MODEL_LOADING"] = "true"
-
-@pytest.fixture(autouse=True)
-def skip_model_loading():
-    """Automatically set environment variable to skip model loading for all tests."""
-    # Environment variable is already set above
-    yield
-    # Keep the variable set for the entire test session