Add updated requirements to fix model load

app.py

@@ -17,10 +17,11 @@ else:
     from transformers import AutoModel, AutoTokenizer
     tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True)
     try:
-        model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True, low_cpu_mem_usage=True, pad_token_id=tokenizer.eos_token_id)
+        model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True, use_safetensors=True, low_cpu_mem_usage=True, pad_token_id=tokenizer.eos_token_id)
         model = model.eval()
     except Exception as e:
         # If model fails to load (e.g. due to no network or heavy resources), keep a placeholder
+        print(f"Warning: Failed to load model: {e}")
         model = None
 
 

requirements.txt

@@ -1,8 +1,68 @@
-gradio
-transformers
-torch
-Pillow
-numpy
-pillow
-numpy
-tiktoken
+aiofiles==24.1.0
+annotated-doc==0.0.4
+annotated-types==0.7.0
+anyio==4.11.0
+brotli==1.2.0
+certifi==2025.11.12
+charset-normalizer==3.4.4
+click==8.3.1
+fastapi==0.121.3
+ffmpy==1.0.0
+filelock==3.20.0
+fsspec==2025.10.0
+gradio==5.50.0
+gradio-client==1.14.0
+groovy==0.1.2
+h11==0.16.0
+hf-xet==1.2.0
+httpcore==1.0.9
+httpx==0.28.1
+huggingface-hub==0.36.0
+idna==3.11
+jinja2==3.1.6
+markdown-it-py==4.0.0
+markupsafe==3.0.3
+mdurl==0.1.2
+mpmath==1.3.0
+networkx==3.6
+numpy==2.3.5
+orjson==3.11.4
+packaging==25.0
+pandas==2.3.3
+pillow==12.0.0
+pydantic==2.12.3
+pydantic-core==2.41.4
+pydub==0.25.1
+pygments==2.19.2
+python-dateutil==2.9.0.post0
+python-multipart==0.0.20
+pytz==2025.2
+pyyaml==6.0.3
+regex==2025.11.3
+requests==2.32.5
+rich==14.2.0
+ruff==0.14.6
+safehttpx==0.1.7
+safetensors==0.7.0
+semantic-version==2.10.0
+setuptools==80.9.0
+shellingham==1.5.4
+six==1.17.0
+sniffio==1.3.1
+starlette==0.50.0
+sympy==1.14.0
+tiktoken==0.12.0
+tokenizers==0.22.1
+tomlkit==0.13.3
+torch==2.9.1
+torchvision==0.24.1
+tqdm==4.67.1
+transformers==4.57.1
+typer==0.20.0
+typing-extensions==4.15.0
+typing-inspection==0.4.2
+tzdata==2025.2
+urllib3==2.5.0
+uvicorn==0.38.0
+verovio==5.6.0
+websockets==15.0.1