Update app.py

app.py

@@ -5,7 +5,7 @@ import torch
 
 st.markdown('## Text-generation OPT from Meta ')
 
-@st.cache(allow_output_mutation=True)
+@st.cache(allow_output_mutation=True,show_spinner=False)
 def get_model():
     return pipeline('text-generation', model=model, skip_special_tokens=True)
 
@@ -24,7 +24,7 @@ with col2:
         model = 'facebook/opt-125m'
 
     if select_model:
-        st.warning('Loading Model... (This may take a while)')
+        with st.spinner('Loading Model... (This may take a while)')
         generator = get_model()    
         st.success('Model loaded correctly!')