Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ st.set_page_config(page_title='Smoke & Fire Detection')
|
|
| 9 |
if 'history' not in st.session_state:
|
| 10 |
st.session_state.history = []
|
| 11 |
|
| 12 |
-
@st.cache(persist=True)
|
| 13 |
def loadModel():
|
| 14 |
pipeline = pip(task="image-classification", model="EdBianchi/vit-fire-detection")
|
| 15 |
return pipeline
|
|
@@ -59,8 +59,8 @@ The fine-tuned model is hosted on the [Hugging Face Hub](https://huggingface.co/
|
|
| 59 |
|
| 60 |
The dataset for fine-tuning process was custom made from different datasets, in particular:
|
| 61 |
|
| 62 |
-
- Samples from "train_fire" and samples from "train_smoke" from https://www.kaggle.com/datasets/kutaykutlu/forest-fire?select=train_fire
|
| 63 |
-
- All the samples (mixed together from further splitting) from https://www.kaggle.com/datasets/mohnishsaiprasad/forest-fire-images
|
| 64 |
|
| 65 |
The custom dataset is hosted on the [Hugging Face Hub](https://huggingface.co/datasets/EdBianchi/SmokeFire).
|
| 66 |
""")
|
|
|
|
| 9 |
if 'history' not in st.session_state:
|
| 10 |
st.session_state.history = []
|
| 11 |
|
| 12 |
+
@st.cache(persist=True, allow_output_mutation=True)
|
| 13 |
def loadModel():
|
| 14 |
pipeline = pip(task="image-classification", model="EdBianchi/vit-fire-detection")
|
| 15 |
return pipeline
|
|
|
|
| 59 |
|
| 60 |
The dataset for fine-tuning process was custom made from different datasets, in particular:
|
| 61 |
|
| 62 |
+
- Samples from "train_fire" and samples from "train_smoke" from [forest-fire dataset](https://www.kaggle.com/datasets/kutaykutlu/forest-fire?select=train_fire).
|
| 63 |
+
- All the samples (mixed together from further splitting) from [forest-fire-images dataset](https://www.kaggle.com/datasets/mohnishsaiprasad/forest-fire-images).
|
| 64 |
|
| 65 |
The custom dataset is hosted on the [Hugging Face Hub](https://huggingface.co/datasets/EdBianchi/SmokeFire).
|
| 66 |
""")
|