Please please work.

app.py

@@ -78,24 +78,48 @@ def load_model(repo_id: str, revision: str = None):
     Works with any public repo_id.
     """
     try:
-        # Clean up revision input (handle empty strings)
+        # Clean up inputs
+        repo_id = repo_id.strip()
+        if not repo_id:
+            return "Please enter a model repo ID"
+            
         if revision and revision.strip() == "":
             revision = None
-            
-        # Load model and processor directly (they handle caching automatically)
-        model = AutoModel.from_pretrained(
-            repo_id, 
-            revision=revision,
-            cache_dir="./model_cache",
-            trust_remote_code=True  # Some models might need this
-        )
         
-        processor = AutoImageProcessor.from_pretrained(
-            repo_id, 
-            revision=revision,
-            cache_dir="./model_cache",
-            trust_remote_code=True
-        )
+        # First try without cache_dir to avoid permission issues
+        try:
+            model = AutoModel.from_pretrained(
+                repo_id, 
+                revision=revision,
+                trust_remote_code=True,
+                use_auth_token=False  # Explicitly no auth for public models
+            )
+            
+            processor = AutoImageProcessor.from_pretrained(
+                repo_id, 
+                revision=revision,
+                trust_remote_code=True,
+                use_auth_token=False
+            )
+        except Exception as e1:
+            # If that fails, try with explicit cache directory
+            model = AutoModel.from_pretrained(
+                repo_id, 
+                revision=revision,
+                cache_dir="/tmp/model_cache",  # Use /tmp for better permissions
+                trust_remote_code=True,
+                use_auth_token=False,
+                local_files_only=False  # Ensure we can download
+            )
+            
+            processor = AutoImageProcessor.from_pretrained(
+                repo_id, 
+                revision=revision,
+                cache_dir="/tmp/model_cache",
+                trust_remote_code=True,
+                use_auth_token=False,
+                local_files_only=False
+            )
 
         # Move to appropriate device
         device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -104,7 +128,7 @@ def load_model(repo_id: str, revision: str = None):
         
         # Validate it's a Vision Transformer
         if not hasattr(model.config, 'patch_size'):
-            return f"❌ Model '{repo_id}' doesn't appear to be a Vision Transformer (no patch_size in config)"
+            return f"Model '{repo_id}' doesn't appear to be a Vision Transformer (no patch_size in config)"
         
         # Update global state
         state["model"] = model
@@ -112,17 +136,19 @@ def load_model(repo_id: str, revision: str = None):
         state["repo_id"] = repo_id
         state["model_type"] = "custom"
         
-        return f"✅ Successfully loaded model '{repo_id}' on {device}"
+        patch_size = model.config.patch_size
+        return f"Successfully loaded ViT model '{repo_id}' (patch size: {patch_size}) on {device}"
         
-    except OSError as e:
-        if "Repository not found" in str(e):
-            return f"❌ Repository '{repo_id}' not found. Please check the repo ID."
-        elif "offline" in str(e).lower():
-            return f"❌ Network error. Please check your internet connection."
-        else:
-            return f"❌ Error accessing model: {str(e)}"
     except Exception as e:
-        return f"❌ Error loading model: {str(e)}"
+        error_str = str(e).lower()
+        if "repository not found" in error_str or "404" in error_str:
+            return f"Repository '{repo_id}' not found. Please check the repo ID."
+        elif "connection" in error_str or "network" in error_str or "offline" in error_str:
+            return f"Network error: {str(e)}"
+        elif "permission" in error_str or "forbidden" in error_str:
+            return f"Permission denied. This might be a private repository."
+        else:
+            return f"Error loading model: {str(e)}"
 
 def display_image(image: Image):
     """