Update app.py

app.py

@@ -58,13 +58,7 @@ def create_model_repo(api, user, orgs_name, model_name, make_private=False):
 
 # ---------------------- MODEL LOADING AND CONVERSION ----------------------
 def download_model(model_path_or_url):
-    """Downloads a model from a URL or Hugging Face Hub, handling various cases.
-    Args:
-        model_path_or_url: Can be a local path, a URL, a Hugging Face repo ID,
-            or a repo ID with a filename (e.g., "user/repo/file.safetensors").
-    Returns:
-        The local path to the downloaded (or existing) file.
-    """
+    """Downloads a model, handling URLs, HF repos, and local paths."""
     try:
         # Check if it's a valid Hugging Face repo ID (and potentially a file within)
         try:
@@ -73,30 +67,29 @@ def download_model(model_path_or_url):
             local_path = hf_hub_download(repo_id=model_path_or_url)
             return local_path
         except HFValidationError:
-            # Not a simple repo ID.  Might be a repo ID with a filename, or a URL.
-            pass
+            pass  # Not a simple repo ID.  Might be repo ID + filename, or a URL.
 
         if model_path_or_url.startswith("http://") or model_path_or_url.startswith("https://"):
-            # It's a URL, use hf_hub_download to handle it (it handles URLs gracefully).
-            local_path = hf_hub_download(repo_id=None, filename=None, repo_type=None, url=model_path_or_url)
+            # It's a URL: use hf_hub_download with the url parameter
+            local_path = hf_hub_download(url=model_path_or_url)  # Corrected line
             return local_path
-        elif os.path.isfile(model_path_or_url): # Local File
+        elif os.path.isfile(model_path_or_url):
+            # It's a local file
             return model_path_or_url
-        else: #HuggingFace Model
-             # Try splitting into repo ID and filename (for "user/repo/file.safetensors")
+        else:
+            # Try splitting into repo ID and filename
             try:
-                parts = model_path_or_url.split("/", 1)  # Split only on the first /
+                parts = model_path_or_url.split("/", 1)
                 if len(parts) == 2:
                     repo_id, filename = parts
-                    validate_repo_id(repo_id) # Check the repo_id part.
+                    validate_repo_id(repo_id)
                     local_path = hf_hub_download(repo_id=repo_id, filename=filename)
                     return local_path
                 else:
-                    raise ValueError("Invalid input")
-            except HFValidationError: #Still invalid
+                    raise ValueError("Invalid input format")
+            except HFValidationError:
                 raise ValueError(f"Invalid model path or URL: {model_path_or_url}")
 
-
     except Exception as e:
         raise ValueError(f"Error downloading or accessing model: {e}")