Update app.py

app.py

@@ -16,25 +16,20 @@ from apscheduler.schedulers.background import BackgroundScheduler
 from textwrap import dedent
 
 HOME = os.environ.get("HOME")
-token = os.environ.get("HF_TOKEN")
+# token = os.environ.get("HF_TOKEN")
 library_username = os.environ.get("OLLAMA_USERNAME").lower()
 ollama_pubkey = open(f"{HOME}/.ollama/id_ed25519.pub", "r")
-token: gr.OAuthToken.token | None
-
-
-
-
-def ollamafy_model(model_id, ollama_q_method, latest, maintainer, ollama_library_username, token):
 
+def ollamafy_model(model_id, ollama_q_method, latest, maintainer, ollama_library_username, auth_token: gr.OAuthToken | None):
     ollama_library_username: library_username | None 
-    if token is None:
+    if oauth_token.token is None:
         raise ValueError("You must be logged in to use Ollamafy")
     # username = whoami(oauth_token.token)["name"]
     model_name = model_id.split('/')[-1]
     fp16 = f"{model_name}-fp16.gguf"
 
     try:
-        api = HfApi(token)   
+        api = HfApi(token=oauth_token.token)   
         dl_pattern = ["*.md", "*.json", "*.model"]
         pattern = (
             "*.safetensors"
@@ -218,7 +213,8 @@ with gr.Blocks(css=css) as demo:
     
 def restart_space():
     ollama_pubkey.close(),
-    HfApi().restart_space(repo_id="unclemusclez/ollamafy", token=HF_TOKEN, library_username=OLLAMA_USERNAME, factory_reboot=True)
+    # HfApi().restart_space(repo_id="unclemusclez/ollamafy", token=HF_TOKEN, library_username=OLLAMA_USERNAME, factory_reboot=True)
+    HfApi().restart_space(repo_id="unclemusclez/ollamafy", factory_reboot=True)
 
 scheduler = BackgroundScheduler()
 scheduler.add_job(restart_space, "interval", seconds=21600)