Spaces:
Runtime error
Runtime error
import streamlit as st | |
from transformers import pipeline, PegasusForConditionalGeneration, PegasusTokenizer | |
from fill_in_summary import FillInSummary | |
from paraphrase import PegasusParaphraser | |
def paraphrase(text): | |
return text | |
st.title('Question Generator by Eddevs') | |
select = st.selectbox('Type', ['Question Generator', 'Paraphrasing', 'Summarization', 'Fill in the blank']) | |
if select == "Question Generator": | |
with st.form("question_gen"): | |
# left_column, right_column = st.columns(2) | |
# left_column.selectbox('Type', ['Question Generator', 'Paraphrasing']) | |
#st.selectbox('Model', ['T5', 'GPT Neo-X']) | |
text_input = st.text_area("Input Text") | |
submitted = st.form_submit_button("Generate") | |
if submitted: | |
with st.spinner('Wait for it...'): | |
result = FillInSummary().summarize(text_input) | |
st.write(text_input) | |
if select == "Summarization": | |
with st.form("summarization"): | |
# left_column, right_column = st.columns(2) | |
# left_column.selectbox('Type', ['Question Generator', 'Paraphrasing']) | |
#st.selectbox('Model', ['T5', 'GPT Neo-X']) | |
text_input = st.text_area("Input Text") | |
submitted = st.form_submit_button("Generate") | |
if submitted: | |
with st.spinner('Wait for it...'): | |
result = FillInSummary().summarize(text_input) | |
st.write(text_input) | |
if select == "Fill in the blank": | |
with st.form("fill_in_the_blank"): | |
text_input = st.text_area("Input Text") | |
submitted = st.form_submit_button("Generate") | |
if submitted: | |
with st.spinner('Wait for it...'): | |
fill = FillInSummary() | |
result = fill.summarize(text_input) | |
result = fill.blank_ne_out(result) | |
st.write(result) | |
if select == "Paraphrasing": | |
with st.form("paraphrasing"): | |
# st.selectbox('Model', ['T5', 'GPT Neo-X']) | |
left_column, right_column = st.columns(2) | |
count = left_column.slider('Count', 0, 10, 3) | |
temperature = right_column.slider('Temperature', 0.0, 10.0, 1.5) | |
text_input = st.text_area("Input Text") | |
submitted = st.form_submit_button("Generate") | |
if submitted: | |
with st.spinner('Wait for it...'): | |
paraphrase_model = PegasusParaphraser(num_return_sequences=count,temperature=temperature) | |
result = paraphrase_model.paraphrase(text_input) | |
st.write(result) | |
#if st.button('Generate'): | |
# st.write(input) | |
#st.success("We have generated 105 Questions for you") | |
# st.snow() | |
##else: | |
##nothing here | |