|
import streamlit as st |
|
|
|
from run_eval import run_generate |
|
|
|
st.write("# GupShup") |
|
st.write("## Summarizing Open-Domain Code-Switched Conversations") |
|
task_type = st.sidebar.selectbox( |
|
"Task type", ["Hinglish to English", "English to English"] |
|
) |
|
model_name = st.sidebar.selectbox( |
|
"Model", ["Pegasus", "mBart", "Bart", "GPT", "T5", "T5_MTL"] |
|
) |
|
|
|
|
|
default_value = "summarize: Debanjan: Hi! kya tum EMNLP attend kar rahein ho? Anish: nehi I will skip this year. what about you? Debanjan: mein virtually attend kar raha hu Debanjan: hopefully can attend the conference in person from next year... Anish: haan dekhtey hain, in person attend karney mein alaag hi maaza hain Debanjan: but I also like this hybrid mode allows to attend the sessions in case you do not make it Anish: right Anish: if you are attending to mujhe interesting papers kuch bhej dena Debanjan: sure" |
|
|
|
conv_form = st.sidebar.form(key="conv_form") |
|
conv_ip = conv_form.text_input(label="Please enter the conversastion", value=default_value) |
|
conv_submit_button = conv_form.form_submit_button(label="Submit") |
|
|
|
|
|
|
|
st.write("### Task Type:", task_type) |
|
st.write("### Model:", model_name) |
|
x = "fg" |
|
|
|
src_file = None |
|
tar_file = None |
|
gen_file = "generated_summary.txt" |
|
score_file = None |
|
|
|
if conv_submit_button: |
|
if conv_ip == "": |
|
st.write("Pls enter non empty conversastion") |
|
else: |
|
st.write("### Summarizing below Conversastion") |
|
st.write(conv_ip) |
|
src_file = "conversastion.txt" |
|
src_fp = open(src_file, "w") |
|
src_fp.write(conv_ip) |
|
src_fp.close() |
|
|
|
tt = "h2e" |
|
if task_type == "English to English": |
|
tt = "e2e" |
|
elif task_type == "Hinglish to English": |
|
tt = "h2e" |
|
model_name_path = "midas/gupshup_" + str(tt) + "_" + str(model_name).lower() |
|
|
|
|
|
result = run_generate( |
|
verbose=True, |
|
model_name_path=model_name_path, |
|
src_txt=src_file, |
|
tar_txt=tar_file, |
|
gen_path=gen_file, |
|
scor_path=score_file, |
|
batch_size=8, |
|
) |
|
|
|
if conv_submit_button: |
|
st.write("summary:") |
|
fp = open(gen_file, "r") |
|
summary = fp.readlines() |
|
fp.close() |
|
st.write(summary) |
|
|
