printing extracted text22

HTR/app.py

@@ -17,8 +17,17 @@ from HTR.strike import struck_images
 from HTR.hcr import text
 from HTR.spell_and_gramer_check import spell_grammer
 
-# Use direct path to local model
-model_path = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), 'models', 'vit-base-beans')
+# Get absolute path to project root
+project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+model_path = os.path.join(project_root, 'models', 'vit-base-beans')
+
+# Log model path for debugging
+logger.info(f"Using model path: {model_path}")
+if not os.path.exists(model_path):
+    logger.error(f"Model directory not found at: {model_path}")
+else:
+    files = os.listdir(model_path)
+    logger.info(f"Found model files: {files}")
 
 def preprocess_image(img):
     """Preprocess image to improve text detection"""

all_models.py

@@ -271,52 +271,69 @@ class ModelSingleton:
             raise
 
     def get_vit_model(self):
-        """Get ViT model with reference counting"""
+        """Get ViT model using only local files - no downloads"""
         try:
             if self.vit_model is None:
-                from transformers import ViTImageProcessor, ViTForImageClassification
-                logger.info("Loading local ViT model...")
+                from transformers import ViTConfig, ViTImageProcessor, ViTForImageClassification
+                logger.info("Loading local ViT model from files...")
                 
-                # Use direct path to local model
-                model_path = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), 'models', 'vit-base-beans')
-                logger.info(f"Using local model path: {model_path}")
+                # Get absolute path to model directory
+                model_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+                model_path = os.path.join(model_root, 'models', 'vit-base-beans')
+                logger.info(f"Using local model directory: {model_path}")
                 
+                # Check model directory exists
                 if not os.path.exists(model_path):
                     raise FileNotFoundError(f"Local model directory not found at: {model_path}")
                 
-                # Check for required files
-                required_files = ['model.safetensors', 'config.json']
-                missing_files = [f for f in required_files if not os.path.exists(os.path.join(model_path, f))]
-                if missing_files:
-                    raise FileNotFoundError(f"Missing required model files: {missing_files}")
+                # Get paths to required files
+                model_file = os.path.join(model_path, 'model.safetensors')
+                config_file = os.path.join(model_path, 'config.json')
                 
-                # Create processor with default settings (no download)
+                # Verify files exist
+                if not os.path.exists(model_file):
+                    raise FileNotFoundError(f"Local model weights file not found at: {model_file}")
+                if not os.path.exists(config_file):
+                    raise FileNotFoundError(f"Local model config file not found at: {config_file}")
+                
+                logger.info("Found all required local model files:")
+                logger.info(f"- Using model weights: {model_file}")
+                logger.info(f"- Using config file: {config_file}")
+                
+                # Load config directly from file
+                logger.info("Loading model configuration from local file...")
+                config = ViTConfig.from_json_file(config_file)
+                
+                # Create processor from local config
+                logger.info("Creating image processor from local config...")
                 self.vit_processor = ViTImageProcessor(
                     do_resize=True,
-                    size=224,
-                    do_normalize=True,
-                    image_mean=[0.5, 0.5, 0.5],
-                    image_std=[0.5, 0.5, 0.5]
+                    size=config.image_size,
+                    do_normalize=True
                 )
                 
-                # Load model from local files only
-                logger.info("Loading ViT model from local files...")
+                # Load model directly from local files
+                logger.info("Loading model weights from local file...")
                 self.vit_model = ViTForImageClassification.from_pretrained(
                     model_path,
-                    local_files_only=True,  # Prevent downloads
-                    use_safetensors=True,   # Use .safetensors file
-                    trust_remote_code=False  # Don't trust or download remote code
+                    config=config,
+                    local_files_only=True,
+                    use_safetensors=True,
+                    trust_remote_code=False,
+                    from_tf=False,
+                    _fast_init=True
                 )
                 
-                # Move model to device
+                logger.info(f"Moving model to {self.device}...")
                 self.vit_model.to(self.device)
                 self.vit_model.eval()
-                logger.info(f"Successfully loaded local ViT model and moved to {self.device}")
+                logger.info("Local model loaded successfully!")
             
             self._reference_counts['vit'] += 1
             return self.vit_model, self.vit_processor
+            
         except Exception as e:
-            logger.error(f"Error loading local ViT model: {e}")
+            logger.error(f"Error loading local ViT model: {str(e)}")
             raise
 
     def release_similarity_model(self):

main.py

@@ -52,6 +52,11 @@ def get_user_cache_dir():
 BASE_DIR = get_user_cache_dir()
 log_print(f"Using base directory: {BASE_DIR}")
 
+# Get absolute path to project root for models
+PROJECT_ROOT = os.path.dirname(os.path.abspath(__file__))
+os.environ['MODEL_ROOT'] = PROJECT_ROOT
+log_print(f"Set MODEL_ROOT to: {PROJECT_ROOT}")
+
 # Set environment variables before any other imports
 cache_dirs = {
     'root': BASE_DIR,
@@ -65,7 +70,8 @@ cache_dirs = {
     'logs': os.path.join(BASE_DIR, 'logs'),
     'uploads': os.path.join(BASE_DIR, 'uploads'),
     'images': os.path.join(BASE_DIR, 'images'),
-    'ans_image': os.path.join(BASE_DIR, 'ans_image')
+    'ans_image': os.path.join(BASE_DIR, 'ans_image'),
+    'models': os.path.join(PROJECT_ROOT, 'models')  # Add models directory
 }
 
 # Create all necessary directories with proper permissions