updates

pages/2_😈_BlackBox_and_WhiteBox_Attacks.py

@@ -219,13 +219,13 @@ if button('Tokenize', key='tokenizer'):
                 if not platform.system().lower() == 'darwin':
                     generator1 = pipeline('text-generation', model='EleutherAI/gpt-neo-1.3B')
             with st.expander('**Original input text**: '+ title):
-                    output_openai = get_codex_response(title)
                     st.markdown(f'The response of GPT-2 with the prompt :blue[{title}]')
                     st.markdown('<blockquote>' + generator(title, max_length=30, num_return_sequences=1)[0]['generated_text'].replace(title, '', 1) + '</blockquote>', unsafe_allow_html=True)
                     if not platform.system().lower() == 'darwin':
-                        st.markdown(f'The response of {model_choice} with the prompt :blue[{title}]')
+                        st.markdown(f'The response of EleutherAI/gpt-neo-1.3B with the prompt :blue[{title}]')
                         st.markdown('<blockquote>' + generator1(title, do_sample=True, min_length=50)[0]['generated_text'].replace(title, '', 1) + '</blockquote>', unsafe_allow_html=True)
 
+                    output_openai = get_codex_response(title)
                     st.markdown(f'The response of [Codex](https://openai.com/blog/openai-codex/) with the prompt :blue[{title}]')
                     st.markdown('<blockquote>' + output_openai + '</blockquote>', unsafe_allow_html=True)
 
@@ -260,7 +260,13 @@ if button('Tokenize', key='tokenizer'):
             st.success(f'We found {restarts} replacements!', icon="✅")
             # st.markdown('<br>'.join(outputs), unsafe_allow_html=True)
             for i in range(restarts):
-                with st.expander(outputs[i]):
+                with st.expander(outputs[i])
+                    st.markdown(f'The response of GPT-2 with the prompt :blue[{outputs[i]}]')
+                    st.markdown('<blockquote>' + generator(outputs[i], max_length=30, num_return_sequences=1)[0]['generated_text'].replace(title, '', 1) + '</blockquote>', unsafe_allow_html=True)
+                    if not platform.system().lower() == 'darwin':
+                        st.markdown(f'The response of EleutherAI/gpt-neo-1.3B with the prompt :blue[{outputs[i]}]')
+                        st.markdown('<blockquote>' + generator1(outputs[i], do_sample=True, min_length=50)[0]['generated_text'].replace(title, '', 1) + '</blockquote>', unsafe_allow_html=True)
+:
                     output_openai = get_codex_response(outputs[i])
                     st.markdown(f'The response of [Codex](https://openai.com/blog/openai-codex/) with the prompt :blue[{outputs[i]}]')
                     st.markdown('<blockquote>' + output_openai + '</blockquote>', unsafe_allow_html=True)