boda's picture
add MLP models
d7e4f1f
import streamlit as st
from model.generate import generate_names
col1, col2, col3 = st.columns([1,1,1])
st.markdown('''This is a simple MLP that predicts the next character given the context of the previous 5 characters!
If you want to play with this, see the repo [here](https://github.com/BodaSadalla98/Arabic_makemore)
''')
option = col2.selectbox(
'Select the language to generate the names with:',
('Arabic', 'English'))
number = col2.number_input(label='Enter how many names to generate..', min_value=0, max_value=1000, value=5)
names = ''
if col2.button('Generate'):
if option =='English':
names = generate_names(number,'en')
else:
names = generate_names(number,'ar')
for name in names:
new_title = f'<p style="font-family:sans-serif;text-align: center; color:Purple; font-size: 30px;">{name}</p>'
col2.markdown(new_title, unsafe_allow_html=True)
st.markdown('''
If you have any questions, contact me: **@bodasadallah**
''')
# st.write(names)