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 |
""")
|