MCP_Prueba / server.py
Cosio's picture
Upload 2 files
9628f3b verified
raw
history blame contribute delete
647 Bytes
from transformers import pipeline
import gradio as gr
# Cargar modelo de Hugging Face
sentiment_pipeline = pipeline(
"sentiment-analysis",
model="distilbert-base-uncased-finetuned-sst-2-english"
)
def sentiment_analysis(text: str) -> dict:
result = sentiment_pipeline(text)[0]
return {
"sentiment": result["label"],
"confidence": round(result["score"], 4)
}
demo = gr.Interface(
fn=sentiment_analysis,
inputs=gr.Textbox(placeholder="Enter text..."),
outputs=gr.JSON(),
title="LLM-based Sentiment Analysis"
)
if __name__ == "__main__":
demo.launch(mcp_server=True)