Upload multi_task_ai_tool.py

multi_task_ai_tool.py

@@ -0,0 +1,36 @@
+
+import streamlit as st
+from transformers import pipeline
+
+# Setting up the Streamlit app
+st.title("Multi-Task AI Tool with Hugging Face")
+st.write("Welcome to the Multi-Task AI Tool! This app can perform sentiment analysis, named entity recognition, and text generation.")
+
+# User input
+user_input = st.text_input("Enter your text below:")
+
+if user_input:
+    # Sentiment Analysis
+    classifier = pipeline('sentiment-analysis')
+    sentiment = classifier(user_input)
+    st.subheader("Sentiment Analysis:")
+    st.write(f"Label: {sentiment[0]['label']}")
+    st.write(f"Confidence: {sentiment[0]['score']:.2f}")
+
+    # Named Entity Recognition
+    ner = pipeline('ner', grouped_entities=True)
+    entities = ner(user_input)
+    st.subheader("Named Entity Recognition:")
+    if entities:
+        for entity in entities:
+            st.write(f"Entity: {entity['entity_group']} - Word: {entity['word']} - Confidence: {entity['score']:.2f}")
+    else:
+        st.write("No named entities found.")
+
+    # Text Generation
+    text_generator = pipeline('text-generation')
+    response = text_generator(user_input, max_length=50, num_return_sequences=1)
+    st.subheader("Text Generation:")
+    st.write(response[0]['generated_text'])
+
+st.write("Type something and see how the AI responds!")