Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from transformers import pipeline | |
| translator = pipeline("translation_en_to_de",model="Helsinki-NLP/opus-mt-en-fr") | |
| sa = pipeline("sentiment-analysis",model = "MarieAngeA13/Sentiment-Analysis-BERT") | |
| text_gen = pipeline("text-generation", model="gpt2") | |
| def translate_to_fr(text): | |
| translate = translator(text ,max_length = len(text.split())+5) | |
| return translate[0]['translation_text'] | |
| def generate_text(prompt): | |
| generated_text = text_gen(prompt, max_length=len(prompt.split()) + 5, num_return_sequences=1, do_sample=True) | |
| return generated_text[0]['generated_text'] | |
| def sentiment_analysis(text): | |
| sentiment = sa(text) | |
| if sentiment[0]['label'] == 'positive': | |
| return "Happy" | |
| if sentiment[0]['label'] == 'negative': | |
| return "Unhappy" | |
| if sentiment[0]['label'] == 'neutral': | |
| return "Neither happy nor unhappy" | |
| with gr.Blocks() as demo: | |
| gr.Markdown("Text Pipeline :Translation, Text Generation, Sentiment Analysis") | |
| with gr.Row(): | |
| translate_btn = gr.Button("Translate") | |
| generate_btn = gr.Button("Generate") | |
| analyze_btn = gr.Button("Analyze") | |
| input_text = gr.Textbox(label="Enter text") | |
| output_text = gr.Textbox(label="Output") | |
| translate_btn.click(translate_to_fr, inputs=input_text, outputs=output_text) | |
| generate_btn.click(generate_text, inputs=input_text, outputs=output_text) | |
| analyze_btn.click(sentiment_analysis, inputs=input_text, outputs=output_text) | |
| demo.launch() |