Update app.py

app.py

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