Upload folder using huggingface_hub

Dockerfile

@@ -17,4 +17,4 @@ USER appuser
 EXPOSE 7860
 ENV HOST=0.0.0.0 PORT=7860 PYTHONUNBUFFERED=1
 
-CMD ["gunicorn", "-k", "uvicorn.workers.UvicornWorker", "main.model_inference:inference_api", "--bind", "0.0.0.0:7860"]
+CMD ["gunicorn", "-k", "uvicorn.workers.UvicornWorker", "main.inference:inference_api", "--bind", "0.0.0.0:7860"]

main/inference.py

@@ -1,6 +1,6 @@
 from fastapi import FastAPI
 from fastapi.responses import JSONResponse
-from hf_serve_api.main.schema import InputData, APIResponse
+from main.schema import InputData, APIResponse
 from datetime import datetime
 from main.utils import *
 import uuid, time