Add import statement for os module

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+
 from load_base_model_data import load_data
 
 ds = load_data()

load_base_model_data.py

@@ -1,3 +1,4 @@
+import os
 from datetime import datetime
 from functools import lru_cache
 from typing import List, Optional
@@ -7,7 +8,6 @@ from huggingface_hub import list_models
 from pydantic import BaseModel, field_validator
 from toolz import groupby
 from tqdm.auto import tqdm
-import os
 
 HF_TOKEN = os.environ.get("HF_TOKEN")