INNER CODE UNIT · Python

forward

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

    def forward(self, c, n):
        embeddings = [self.embs[i](c[:, i]) for i in range(len(self.embs))]
        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)

View source record →