Update app.py

app.py

@@ -7,14 +7,14 @@ st.markdown('## Text-generation OPT from Meta ')
 
 @st.cache(allow_output_mutation=True)
 def get_model():
-    return pipeline('text-generation', model=model)
+    return pipeline('text-generation', model=model, skip_special_tokens=True)
 
 col1, col2 = st.beta_columns([2,1])
 
 with col2:
     select_model = st.radio(
         "Select the model to use:",
-        ('OPT-125m', 'OPT-350m', 'OPT-1.3b'))
+        ('OPT-125m', 'OPT-350m', 'OPT-1.3b'), index = 1)
 
     if select_model == 'OPT-1.3b':
         model = 'facebook/opt-1.3b'
@@ -23,11 +23,9 @@ with col2:
     elif select_model == 'OPT-125m':
         model = 'facebook/opt-125m'
 
-    if select_model:
-        with st.spinner('Loading Model... (This may take a while)'):
-            generator = get_model()    
-            #time.sleep(2)
-            st.success('Model loaded correctly!')
+    with st.warning('Loading Model... (This may take a while)'):
+        generator = get_model()    
+        st.success('Model loaded correctly!')
     
 
 with col1: