File size: 922 Bytes
ebd8530
 
 
993705a
ebd8530
3319852
577ffe8
ebd8530
577ffe8
 
ebd8530
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
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'])