SHRI171981 commited on
Commit ·
2fe87d5
0
Parent(s):
Pushed HF files
Browse files- .github/workflows/sync_to_hub.yml +19 -0
- README.md +12 -0
- app.py +74 -0
- gitattributes +35 -0
- requirements.txt +5 -0
.github/workflows/sync_to_hub.yml
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: Sync to Hugging Face Hub
|
| 2 |
+
|
| 3 |
+
on:
|
| 4 |
+
push:
|
| 5 |
+
branches: [main]
|
| 6 |
+
|
| 7 |
+
# This workflow runs whenever you push to the main branch
|
| 8 |
+
jobs:
|
| 9 |
+
sync-to-hub:
|
| 10 |
+
runs-on: ubuntu-latest
|
| 11 |
+
steps:
|
| 12 |
+
- uses: actions/checkout@v3
|
| 13 |
+
with:
|
| 14 |
+
fetch-depth: 0
|
| 15 |
+
lfs: true
|
| 16 |
+
- name: Push to hub
|
| 17 |
+
env:
|
| 18 |
+
HF_TOKEN: ${{ secrets.HF_TOKEN }}
|
| 19 |
+
run: git push https://shri171981:$HF_TOKEN@huggingface.co/spaces/shri171981/genai_sentiment_classifier main
|
README.md
ADDED
|
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
title: Genai Sentiment Classifier
|
| 3 |
+
emoji: 🐠
|
| 4 |
+
colorFrom: indigo
|
| 5 |
+
colorTo: blue
|
| 6 |
+
sdk: gradio
|
| 7 |
+
sdk_version: 6.0.1
|
| 8 |
+
app_file: app.py
|
| 9 |
+
pinned: false
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
+
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
app.py
ADDED
|
@@ -0,0 +1,74 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import gradio as gr
|
| 2 |
+
from transformers import pipeline
|
| 3 |
+
|
| 4 |
+
# 1. SETUP MODEL
|
| 5 |
+
MODEL_ID = "shri171981/genai_project_deployment"
|
| 6 |
+
|
| 7 |
+
# 2. DEFINE LABELS
|
| 8 |
+
LABELS = ["Anger", "Fear", "Joy", "Sadness", "Surprise"]
|
| 9 |
+
|
| 10 |
+
# 3. LOAD PIPELINE
|
| 11 |
+
pipe = pipeline("text-classification", model=MODEL_ID, top_k=None)
|
| 12 |
+
|
| 13 |
+
def predict(text):
|
| 14 |
+
# Run inference
|
| 15 |
+
results = pipe(text)[0]
|
| 16 |
+
|
| 17 |
+
# Create a dictionary of { "Joy": 0.95, "Anger": 0.02 } for Gradio
|
| 18 |
+
output_scores = {}
|
| 19 |
+
for result in results:
|
| 20 |
+
# Extract the ID (e.g., "LABEL_0" -> 0)
|
| 21 |
+
label_id = int(result['label'].split('_')[-1])
|
| 22 |
+
|
| 23 |
+
# Map ID to the text name (e.g., 0 -> "Anger")
|
| 24 |
+
label_name = LABELS[label_id]
|
| 25 |
+
|
| 26 |
+
# Add to dict
|
| 27 |
+
output_scores[label_name] = result['score']
|
| 28 |
+
|
| 29 |
+
return output_scores
|
| 30 |
+
|
| 31 |
+
# 4. UI
|
| 32 |
+
theme = gr.themes.Soft(
|
| 33 |
+
primary_hue="orange",
|
| 34 |
+
secondary_hue="slate",
|
| 35 |
+
)
|
| 36 |
+
|
| 37 |
+
with gr.Blocks() as demo:
|
| 38 |
+
gr.Markdown(
|
| 39 |
+
"""
|
| 40 |
+
# Emotion Classifier
|
| 41 |
+
### Analyze the sentiment of your text across 5 different emotions.
|
| 42 |
+
"""
|
| 43 |
+
)
|
| 44 |
+
|
| 45 |
+
with gr.Row():
|
| 46 |
+
with gr.Column():
|
| 47 |
+
input_text = gr.Textbox(
|
| 48 |
+
label="Input Text",
|
| 49 |
+
placeholder="Type something emotional here...",
|
| 50 |
+
lines=3
|
| 51 |
+
)
|
| 52 |
+
submit_btn = gr.Button("Analyze Emotion", variant="primary")
|
| 53 |
+
|
| 54 |
+
gr.Examples(
|
| 55 |
+
examples=[
|
| 56 |
+
["I can't believe you would betray me like this!"], # Anger
|
| 57 |
+
["I heard a strange noise outside and I'm scared to look."], # Fear
|
| 58 |
+
["I finally got the promotion! This is the best day ever!"], # Joy
|
| 59 |
+
["I feel so lonely and empty inside."], # Sadness
|
| 60 |
+
["Wow! I never expected a surprise party!"] # Surprise
|
| 61 |
+
],
|
| 62 |
+
inputs=input_text
|
| 63 |
+
)
|
| 64 |
+
|
| 65 |
+
with gr.Column():
|
| 66 |
+
# This component creates the bar chart automatically
|
| 67 |
+
output_chart = gr.Label(label="Emotion Confidence Scores", num_top_classes=5)
|
| 68 |
+
|
| 69 |
+
# Link the button to the function
|
| 70 |
+
submit_btn.click(fn=predict, inputs=input_text, outputs=output_chart)
|
| 71 |
+
|
| 72 |
+
# 5. LAUNCH
|
| 73 |
+
if __name__ == "__main__":
|
| 74 |
+
demo.launch()
|
gitattributes
ADDED
|
@@ -0,0 +1,35 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
*.7z filter=lfs diff=lfs merge=lfs -text
|
| 2 |
+
*.arrow filter=lfs diff=lfs merge=lfs -text
|
| 3 |
+
*.bin filter=lfs diff=lfs merge=lfs -text
|
| 4 |
+
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
| 5 |
+
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
| 6 |
+
*.ftz filter=lfs diff=lfs merge=lfs -text
|
| 7 |
+
*.gz filter=lfs diff=lfs merge=lfs -text
|
| 8 |
+
*.h5 filter=lfs diff=lfs merge=lfs -text
|
| 9 |
+
*.joblib filter=lfs diff=lfs merge=lfs -text
|
| 10 |
+
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
| 11 |
+
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
| 12 |
+
*.model filter=lfs diff=lfs merge=lfs -text
|
| 13 |
+
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
| 14 |
+
*.npy filter=lfs diff=lfs merge=lfs -text
|
| 15 |
+
*.npz filter=lfs diff=lfs merge=lfs -text
|
| 16 |
+
*.onnx filter=lfs diff=lfs merge=lfs -text
|
| 17 |
+
*.ot filter=lfs diff=lfs merge=lfs -text
|
| 18 |
+
*.parquet filter=lfs diff=lfs merge=lfs -text
|
| 19 |
+
*.pb filter=lfs diff=lfs merge=lfs -text
|
| 20 |
+
*.pickle filter=lfs diff=lfs merge=lfs -text
|
| 21 |
+
*.pkl filter=lfs diff=lfs merge=lfs -text
|
| 22 |
+
*.pt filter=lfs diff=lfs merge=lfs -text
|
| 23 |
+
*.pth filter=lfs diff=lfs merge=lfs -text
|
| 24 |
+
*.rar filter=lfs diff=lfs merge=lfs -text
|
| 25 |
+
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
| 26 |
+
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
| 27 |
+
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
| 28 |
+
*.tar filter=lfs diff=lfs merge=lfs -text
|
| 29 |
+
*.tflite filter=lfs diff=lfs merge=lfs -text
|
| 30 |
+
*.tgz filter=lfs diff=lfs merge=lfs -text
|
| 31 |
+
*.wasm filter=lfs diff=lfs merge=lfs -text
|
| 32 |
+
*.xz filter=lfs diff=lfs merge=lfs -text
|
| 33 |
+
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
+
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
+
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
requirements.txt
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
transformers
|
| 2 |
+
torch
|
| 3 |
+
sentencepiece
|
| 4 |
+
protobuf
|
| 5 |
+
accelerate
|