EdBianchi commited on
Commit
49c0213
1 Parent(s): 279dc99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
  """)