Update app.py

app.py

@@ -17,13 +17,10 @@ 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()
 
-
-
-
-def regenerate_pubkey(login, oauth_token: gr.OAuthToken | None):
+def regenerate_pubkey(auth_token: gr.OAuthToken | None):
     try:
         if oauth_token.token is None:
             raise ValueError("You must be logged in to use Ollamafy")
@@ -56,8 +53,8 @@ def regenerate_pubkey(login, oauth_token: gr.OAuthToken | None):
     except Exception as e:
         return (f"Error: {e}", "error.png")
     finally:
-        shutil.rmtree(model_name, ignore_errors=True)
-        print("Folder cleaned up successfully`!")
+        # shutil.rmtree(model_name, ignore_errors=True)
+        print("Ollama pubkey Generated!")
 
 
 def ollamafy_model(login, account, model_id, ollama_library_username , ollama_q_method, latest, maintainer, oauth_token: gr.OAuthToken | None):
@@ -266,8 +263,7 @@ 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", factory_reboot=True)
+    HfApi().restart_space(repo_id="unclemusclez/ollamafy", token=HF_TOKEN, library_username=OLLAMA_USERNAME, factory_reboot=True)
 
 scheduler = BackgroundScheduler()
 scheduler.add_job(restart_space, "interval", seconds=21600)