Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import pipeline as pip
 from PIL import Image
 
 # set page setting
@@ -11,7 +11,7 @@ if 'history' not in st.session_state:
 
 @st.cache(persist=True)
 def loadModel():
-    pipeline = pipeline(task="image-classification", model="EdBianchi/vit-fire-detection")
+    pipeline = pip(task="image-classification", model="EdBianchi/vit-fire-detection")
     return pipeline
 
 # PROCESSING