Update endpoints.py

endpoints.py

@@ -4,11 +4,14 @@ from PIL import Image
 import urllib.request
 from io import BytesIO
 import json
-import os
 from config import settings  
 import utils
 from routers import inference, training
 from routers.donut_inference import process_document_donut
+from huggingface_hub import login
+import os
+# login(settings.huggingface_key)
+login(os.getenv("HUGGINGFACE_KEY"))
 
 app = FastAPI(openapi_url="/api/v1/sparrow-ml/openapi.json", docs_url="/api/v1/sparrow-ml/docs")
 
@@ -64,8 +67,7 @@ async def run_inference(file: Optional[UploadFile] = File(None), image_url: Opti
 
         processing_time = 0
         if model_in_use == 'donut':
-            result, processing_time = process_document_donut(image, model_url)  # Pass model_url to the function
-        # parse file name from url
+            result, processing_time = process_document_donut(image, model_url)
         file_name = image_url.split("/")[-1]
         utils.log_stats(settings.inference_stats_file, [processing_time, count_values(result), file_name, settings.model])
         print(f"Processing time inference: {processing_time:.2f} seconds")