INNER CODE UNIT · Python

x

Niketkumardheeryan/ML-CaPsule · Air_quality_prediction/app.py:36

        x = torch.cat(embeddings + [n], dim=1)
        x = self.input_layer(x)
        x = x + self.res_block(x)
        return self.head(x)

# --- 2. SETUP & LOAD ARTIFACTS ---
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")

@st.cache_resource
def load_artifacts():
    if not MODEL_PATH.is_file():
        raise FileNotFoundError(
            f"Model artifact not found at {MODEL_PATH}. "
            "Run AQI_pred.ipynb to generate weights.pkl first."
        )
    return joblib.load(MODEL_PATH)

try:

View source record →