krupal02's picture
Update sdk_version to 5.12.0 in README to fix Gradio 5.x conflict
c3f08db
|
Raw
History Blame Contribute Delete
1.4 kB
---
title: Multi-Hazard Warning System
emoji: 🔥
colorFrom: red
colorTo: blue
sdk: gradio
sdk_version: "5.12.0"
app_file: app.py
pinned: true
license: mit
tags:
- wildfire
- aqi
- multi-task-learning
- deep-learning
- pytorch
- resnet50
- bilstm
- gradcam
- air-quality
- fire-detection
short_description: "Wildfire risk + AQI forecasting via MTL"
---
# 🔥 Multi-Hazard Warning System
**Multi-Task Learning for Joint Wildfire Risk Prediction & AQI Forecasting**
A production-grade deep learning system that simultaneously predicts wildfire risk and forecasts Air Quality Index (AQI) for any user-selected region using Multi-Task Learning (MTL).
## 🏗️ Architecture
- **CNN Branch**: ResNet-50 (4-channel: RGB + NIR) → 2048-dim spatial features
- **LSTM Branch**: 2-layer Bidirectional LSTM → 256-dim temporal features
- **Fusion**: Concatenation → Shared FC layers (2304 → 512 → 256)
- **Task 1**: Wildfire Risk Heatmap (128×128 spatial probability map)
- **Task 2**: 72-hour AQI Forecast (hourly PM2.5 predictions)
## 🎯 Features
- Interactive Folium map with fire risk heatmap overlay
- Grad-CAM explainability visualization
- 72-hour AQI forecast chart with WHO thresholds
- Location search via geocoding
- Real-time weather data integration
## 🛠️ Tech Stack
PyTorch · ResNet-50 · BiLSTM · Gradio · FastAPI · Folium · Plotly · OpenCV