changes

README.md

@@ -3,8 +3,8 @@ title: Emotion Classifier (NLP)
 emoji: 🧠
 colorFrom: indigo
 colorTo: pink
-sdk: gradio
-sdk_version: 4.27.0
+sdk: streamlit
+sdk_version: 1.32.2
 app_file: app/app.py
 pinned: false
 ---

app/app.py

@@ -1,6 +1,3 @@
-import os
-os.system("git lfs pull")
-
 import streamlit as st
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
@@ -20,7 +17,7 @@ GOEMOTIONS_LABELS = [
 # -----------------------------
 # Load model and tokenizer
 # -----------------------------
-MODEL_PATH = "outputs/model"
+MODEL_PATH = "./outputs/model"
 model = AutoModelForSequenceClassification.from_pretrained(MODEL_PATH)
 model.to("cpu")
 tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
@@ -41,8 +38,8 @@ if st.button("Classify") and input_text.strip():
         outputs = model(**inputs)
         probs = torch.nn.functional.softmax(outputs.logits, dim=1)[0]
 
-    pred_label_idx = torch.argmax(probs).cpu().item()
-    pred_score = probs[pred_label_idx].cpu().item()
+    pred_label_idx = torch.argmax(probs).item()
+    pred_score = probs[pred_label_idx].item()
     pred_emotion = GOEMOTIONS_LABELS[pred_label_idx]
 
     # -----------------------------