Update app.py

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 import json
 import torch
 from transformers import AutoTokenizer
-from transformers import AutoModelForSequenceClassification
 from model import MultitaskModel
 from huggingface_hub import hf_hub_download
 
@@ -20,14 +19,11 @@ label_emo = [None] * len(emotion2id)
 for k, v in emotion2id.items():
     label_emo[v] = k
 
-# Load model
+# Load tokenizer
 tokenizer = AutoTokenizer.from_pretrained("multitask-nlp/chatbot-multitask-sentiment-emotion")
-model = AutoModelForSequenceClassification.from_pretrained(
-    "multitask-nlp/chatbot-multitask-sentiment-emotion",
-    filename="pytorch_model.bin"
-)
-
 
+# Load model
+model_path = hf_hub_download(repo_id="multitask-nlp/chatbot-multitask-sentiment-emotion", filename="pytorch_model.bin")
 model = MultitaskModel("distilbert-base-uncased", 2, 7)
 model.load_state_dict(torch.load(model_path, map_location="cpu"))
 model.eval()
@@ -55,4 +51,4 @@ demo = gr.Interface(
     description="Masukkan teks untuk memprediksi sentimen dan emosi."
 )
 
-demo.launch() 
+demo.launch()