Added waiting spinner

app.py

@@ -26,10 +26,11 @@ def main():
     # Text input
     user_input = st.text_area("Enter your prompt:", height=300)
     
-    if st.button('Generate'):
-        # Get the model's completion
-        completion = get_completion(user_input)
-        st.text_area("Model Completion:", value=completion, height=300, key="2")
+    with st.spinner('Generating completion...'):
+        if st.button('Generate'):
+            # Get the model's completion
+            completion = get_completion(user_input)
+            st.text_area("Model Completion:", value=completion, height=300, key="2")
 
 if __name__ == '__main__':
     main()