Update app.py

app.py

@@ -6,10 +6,10 @@ import os
 import time
 
 # ------------------------------------------------------
-# 1. Load the Specialized OCR Model
+# 1. Load the CPU-Patched Model
 # ------------------------------------------------------
-# ~0.6B parameters. Much faster than Qwen-2B.
-MODEL_ID = "stepfun-ai/GOT-OCR2_0"
+# This is the specific repo that fixes the "Found no NVIDIA driver" error.
+MODEL_ID = "srimanth-d/GOT_CPU"
 
 print(f"⏳ Loading {MODEL_ID}...")
 
@@ -20,7 +20,7 @@ tokenizer = AutoTokenizer.from_pretrained(
 )
 
 # Load Model
-# low_cpu_mem_usage=True helps load it fast into your 18GB RAM
+# low_cpu_mem_usage=True is safe here because this repo is patched for CPU.
 model = AutoModel.from_pretrained(
     MODEL_ID, 
     trust_remote_code=True, 
@@ -30,7 +30,6 @@ model = AutoModel.from_pretrained(
     pad_token_id=tokenizer.eos_token_id
 )
 
-# Ensure model is in float32 for CPU operations (Half-precision crashes CPUs)
 model = model.eval().float() 
 print(f"✅ {MODEL_ID} Loaded! Ready for handwriting.")