|
import streamlit as st |
|
from streamlit_tags import st_tags, st_tags_sidebar |
|
from keytotext import pipeline |
|
from PIL import Image |
|
|
|
|
|
|
|
|
|
|
|
st.write("# Code for Keywords to Text") |
|
|
|
st.markdown("***Idea is to build a model which will take keywords as inputs and generate information as outputs.***") |
|
image = Image.open('1.png') |
|
st.image(image) |
|
|
|
st.sidebar.write("# Parameter Selection") |
|
maxtags_sidebar = st.sidebar.slider('Number of tags allowed?', 1, 10, 1, key='ehikwegrjifbwreuk') |
|
keywords = st_tags( |
|
label='# Enter Keywords:', |
|
text='Press enter to add more', |
|
value=['Summer'], |
|
suggestions=['five', 'six', 'seven', 'eight', 'nine', 'three', 'eleven', 'ten', 'four'], |
|
maxtags=maxtags_sidebar, |
|
key="aljnf") |
|
|
|
|
|
option = st.sidebar.selectbox( |
|
'Which model would you like to be selected?', |
|
('mrm8488/t5-base-finetuned-common_gen', 'k2t-base', 'k2t')) |
|
|
|
|
|
|
|
|
|
nlp=pipeline(option) |
|
st.sidebar.success("Load Successfully!") |
|
|
|
st.write("## Results:") |
|
if st.button('Generate Sentence'): |
|
out=nlp(keywords) |
|
st.success(out) |
|
|
