support submit gated model

app.py

@@ -3,16 +3,15 @@ import gradio as gr
 import os
 import re
 import requests
-import huggingface_hub
-from huggingface_hub.utils._errors import EntryNotFoundError, RepositoryNotFoundError
 from dotenv import load_dotenv
 from matplotlib.colors import LinearSegmentedColormap
 import plotly.express as px
 import plotly.graph_objects as go
 from sklearn.linear_model import LinearRegression
 import numpy as np
-from huggingface_hub import HfApi, HfFolder
+from huggingface_hub import HfApi
 from huggingface_hub.hf_api import HTTPError
+from huggingface_hub.utils._errors import GatedRepoError
 
 load_dotenv()
 webhook_url = os.environ.get("WEBHOOK_URL")
@@ -228,16 +227,18 @@ def check_model_exists(model_id):
     api = HfApi()
     try:
         model_info = api.model_info(model_id)
-        return True
+        return "Exists and is accessible"
+    except GatedRepoError:
+        return "Exists but is restricted"
     except HTTPError as e:
         if e.response.status_code == 404:
-            return False
+            return "Does not exist"
         else:
-            raise
+            return "Error: " + str(e)
 
 
 def submit_model(name):
-    if not check_model_exists(name):
+    if 'Exists' not in check_model_exists(name):
         return f"# ERROR: Model {name} does not exist on Hugging Face!"
 
     try: