Update app.py

app.py

@@ -27,14 +27,15 @@ 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()    
-            st.success('Model loaded correctly!')
+    with st.spinner('Loading Model... (This may take a while)'):
+        generator = get_model()    
+        st.success('Model loaded correctly!')
      
+gen = st.info('Generating text...')
 answer = generator(prompt,
                        max_length=100,
-                       no_repeat_ngram_size=3, early_stopping=True, num_beams=10, skip_special_tokens=True)
+                       no_repeat_ngram_size=3, early_stopping=True, num_beams=10, skip_special_tokens=True)                      
+gen.empty()                      
                        
 lst = answer[0]['generated_text']