Update app.py

app.py

@@ -9,7 +9,7 @@ st.markdown('## Text-generation OPT from Meta ')
 def get_model():
     return pipeline('text-generation', model=model, skip_special_tokens=True)
 
-col1, col2 = st.beta_columns([2,1])
+col1, col2 = st.columns([2,1])
 
 with col2:
     select_model = st.radio(
@@ -23,16 +23,20 @@ with col2:
     elif select_model == 'OPT-125m':
         model = 'facebook/opt-125m'
 
-    with st.warning('Loading Model... (This may take a while)'):
+    if select_model:
+        st.warning('Loading Model... (This may take a while)'):
         generator = get_model()    
         st.success('Model loaded correctly!')
     
 
 with col1:
     prompt= st.text_area('Your prompt here',
-        '''AI will help humanity?''') 
+        '''AI will help humanity? ''') 
         
-    answer = generator(prompt, max_length=60, no_repeat_ngram_size=2, early_stopping=True, num_beams=5)
+    answer = generator(prompt,
+                       max_length=60,
+                       no_repeat_ngram_size=2, early_stopping=True, num_beams=5, , skip_special_tokens=True)
+                       
     lst = answer[0]['generated_text']
 
     # answer = ['ciao come stai stutto bene']