|
import gradio as gr |
|
from transformers import pipeline |
|
|
|
|
|
title_emotion = "Classify text according to emotion" |
|
description_emotion = "Emotion text classification by Vishal Tiwari " |
|
examples_emotion = [ |
|
["Remember before Twitter when you took a photo of food, got the film developed, then drove around showing everyone the pic? No? Me neither."], |
|
['''"We are all here because we're committed to the biggest question of all: What's out there?" Take your first steps toward answering that question by watching our Gameplay Reveal from the #XboxBethesda Showcase. '''], |
|
["A STUNNER IN KNOXVILLE! 😱 Notre Dame takes down No. 1 Tennessee for its first trip to Omaha in 20 years‼️"], |
|
["you and I best moment is yet to come 💜 #BTS9thAnniversary"] |
|
] |
|
|
|
interface_emotion = gr.Interface.load( |
|
"huggingface/bhadresh-savani/bert-base-go-emotion", |
|
title=title_emotion, |
|
description=description_emotion, |
|
examples=examples_emotion |
|
) |
|
|
|
|
|
title_tts = "Text to Speech Translation" |
|
examples_tts = [ |
|
"I love learning machine learning", |
|
"How do you do?", |
|
] |
|
|
|
interface_tts = gr.Interface.load( |
|
"huggingface/facebook/fastspeech2-en-ljspeech", |
|
title=title_tts, |
|
examples=examples_tts, |
|
description="Give me something to say!", |
|
) |
|
|
|
|
|
demo = gr.TabbedInterface([interface_emotion, interface_tts], ["Emotion Classification", "Text to Speech"]) |
|
|
|
if __name__ == "__main__": |
|
demo.launch() |