alex-abb's picture
Update app.py
cf4674b verified
raw
history blame
875 Bytes
import gradio as gr
from transformers import pipeline
import spaces
# Charger le pipeline de classification des sentiments
sentiment_analyzer = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
@spaces.GPU
def analyze_sentiment(text):
# Analyser le sentiment du texte
result = sentiment_analyzer(text)
# Retourner le résultat sous forme de texte
sentiment = result[0]['label']
score = result[0]['score']
return f"Sentiment: {sentiment}, Score: {score:.2f}"
# Créer l'interface Gradio
interface = gr.Interface(
fn=analyze_sentiment,
inputs="text",
outputs="text",
title="Analyse des Sentiments",
description="Entrez un texte pour analyser son sentiment (positif, négatif ou neutre)."
)
# Lancer l'interface dans un Space Hugging Face
if __name__ == "__main__":
interface.launch()