Upload folder using huggingface_hub

llama.py

@@ -1,7 +1,7 @@
 import subprocess
 import os
 import shutil
-from huggingface_hub.HfApi import upload_folder
+from huggingface_hub import HfApi
 
 
 PARALLEL = "8"
@@ -11,7 +11,8 @@ TEMP_DIR = "llama_build"
 
 
 def upload(folder_path, repo_id, repo_type="model"):
-    upload_folder(
+    api = HfApi()
+    api.upload_folder(
         folder_path=folder_path,
         repo_id=repo_id,
         repo_type=repo_type,

ollama.py

@@ -2,7 +2,7 @@ import subprocess
 import os
 import sys
 import shutil
-from huggingface_hub.HfApi import upload_folder
+from huggingface_hub import HfApi
 
 
 PARALLEL = "8"
@@ -12,7 +12,8 @@ TEMP_DIR = "/tmp/build"
 
 
 def upload(folder_path, repo_id, repo_type="model"):
-    upload_folder(
+    api = HfApi()
+    api.upload_folder(
         folder_path=folder_path,
         repo_id=repo_id,
         repo_type=repo_type,

start.py

@@ -4,7 +4,6 @@ from datetime import datetime, timedelta
 
 from fastapi import FastAPI
 from apscheduler.schedulers.background import BackgroundScheduler
-from huggingface_hub.HfApi import restart_space
 import uvicorn
 
 app = FastAPI()