Matt09Miao's picture
Update app.py
577ffe8 verified
raw
history blame
922 Bytes
import streamlit as st
from transformers import pipeline
st.set_page_config(page_title="Generate Your Tweet and Toxicity Analysis")
st.header("Please input your first word of a Tweet :performing_arts:")
input = st.text_input("In put your first word...")
if input is not None:
print(Your word is welcome :slightly_smiling_face:)
#Stage 1: Image to Text
st.text('Processing img2text...')
scenario = img2text(uploaded_file.name)
st.write(scenario)
#Stage 2: Text to Story
st.text('Generating a story...')
story = text2story(scenario)
st.write(story)
#Stage 3: Story to Audio data
st.text('Generating audio data...')
audio_data =text2audio(story)
# Play button
if st.button("Play Audio"):
st.audio(audio_data['audio'],
format="audio/wav",
start_time=0,
sample_rate = audio_data['sampling_rate'])