Update app.py

app.py

@@ -4,27 +4,31 @@ import warnings
 import gradio as gr
 from vaderSentiment.vaderSentiment import SentimentIntensityAnalyzer
 import traceback
+from huggingface_hub import hf_hub_download
 
 warnings.filterwarnings('ignore')
 
-# Tentukan direktori untuk menyimpan model di direktori lokal repository Hugging Face Spaces
-model_dir = "."  # Direktori kerja di Hugging Face Spaces
+# Tentukan repository dan nama file model di Hugging Face Hub
+repo_id = "renaldidafa/sentimentanalysis"
+model_filename = 'model_DecisionTreeClassifier.pkl'
+vectorizer_filename = 'tfidf_vectorizer.pkl'
+label_encoder_filename = 'label_encoder.pkl'
 
-# Memuat model dari file di dalam direktori yang ditentukan
-model_filename = os.path.join(model_dir, 'model_DecisionTreeClassifier.pkl')
-vectorizer_filename = os.path.join(model_dir, 'tfidf_vectorizer.pkl')
-label_encoder_filename = os.path.join(model_dir, 'label_encoder.pkl')
+# Unduh file model dari Hugging Face Hub
+model_path = hf_hub_download(repo_id=repo_id, filename=model_filename)
+vectorizer_path = hf_hub_download(repo_id=repo_id, filename=vectorizer_filename)
+label_encoder_path = hf_hub_download(repo_id=repo_id, filename=label_encoder_filename)
 
 # Logging untuk memeriksa apakah file ada
 print("Checking if files exist:")
-print("Model file exists:", os.path.exists(model_filename))
-print("Vectorizer file exists:", os.path.exists(vectorizer_filename))
-print("Label encoder file exists:", os.path.exists(label_encoder_filename))
+print("Model file exists:", os.path.exists(model_path))
+print("Vectorizer file exists:", os.path.exists(vectorizer_path))
+print("Label encoder file exists:", os.path.exists(label_encoder_path))
 
 # Memuat model
-loaded_model = joblib.load(model_filename)
-vectorizer = joblib.load(vectorizer_filename)
-label_encoder = joblib.load(label_encoder_filename)
+loaded_model = joblib.load(model_path)
+vectorizer = joblib.load(vectorizer_path)
+label_encoder = joblib.load(label_encoder_path)
 
 # Buat instance dari SentimentIntensityAnalyzer
 analyzer = SentimentIntensityAnalyzer()