demo / app.py
ahmedalrashedi's picture
Update app.py
e8be44f
raw history blame
No virus
756 Bytes
import streamlit as st
from transformers import pipeline
st.title("Sentiment Analysis And Translation App")
text = st.text_input("Enter a sentence:")
sentiment_analyzer = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
translation = pipeline("translation_en_to_ar", model="anibahug/marian-finetuned-kde4-en-to-ar")
if text:
results = sentiment_analyzer(text)
sentiment = results[0]['label']
confidence = results[0]['score']
st.write(f"Sentiment: {sentiment}")
st.write(f"Confidence: {confidence:.2f}")
# if text:
# result = sentiment_analysis(text)
# st.json(result)
if text:
result = translation(text)[0]
st.write(f"Translated text: {result['translation_text']}")