Update app.py

app.py

@@ -15,15 +15,14 @@ from gradio_huggingfacehub_search import HuggingfaceHubSearch
 from apscheduler.schedulers.background import BackgroundScheduler
 from textwrap import dedent
 
-HF_TOKEN = os.environ.get("HF_TOKEN")
-OLLAMA_USERNAME = os.environ.get("OLLAMA_USERNAME").lower()
 HOME = os.environ.get("HOME")
+token = os.environ.get("HF_TOKEN")
+library_username = os.environ.get("OLLAMA_USERNAME").lower()
 ollama_pubkey = open(f"{HOME}/.ollama/id_ed25519.pub", "r")
-oauth_token: gr.OAuthToken | None
-library_username: OLLAMA_USERNAME | None
-token = HF_TOKEN | None
 
-def ollamafy_model(model_id, ollama_q_method, latest, maintainer, oauth_token , library_username):
+
+
+def ollamafy_model(model_id, ollama_q_method, latest, maintainer, library_username: ollama_library_username | None, token = gr.OAuthToken.token | None ):
     if oauth_token.token is None && token is None:
         raise ValueError("You must be logged in to use Ollamafy")
     # username = whoami(oauth_token.token)["name"]
@@ -31,11 +30,6 @@ def ollamafy_model(model_id, ollama_q_method, latest, maintainer, oauth_token ,
     fp16 = f"{model_name}-fp16.gguf"
 
     try:
-        if gr.OAuthToken:
-            oauth_token = gr.OAuthToken
-            token = oauth_token.toke
-        else: token = HF_TOKEN    
-            
         api = HfApi(token):        
         dl_pattern = ["*.md", "*.json", "*.model"]
         pattern = (