Update app.py

app.py

@@ -3,9 +3,33 @@ from transformers import DistilBertTokenizerFast, DistilBertForSequenceClassific
 import torch
 import torch.nn.functional as F
 
+import zipfile
+import shutil
+import os
+
+def unzip_and_save(zip_file_path, extraction_path):
+    # Create the extraction directory if it doesn't exist
+    os.makedirs(extraction_path, exist_ok=True)
+
+    with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
+        folder_name = os.path.basename(zip_file_path).split('.')[0]
+        zip_ref.extractall(extraction_path)
+        source_path = os.path.join(extraction_path, folder_name)
+        destination_path = os.path.join(extraction_path, folder_name)
+        if os.path.exists(destination_path):
+            print(f"Error: Destination path '{destination_path}' already exists")
+        else:
+            shutil.move(source_path, destination_path)
+
+# Example usage:
+zip_file_path = 'finetuned_bert_sentiment_harsh.zip'  # Path to your ZIP file
+extraction_path = 'bert_model_sentiment_v1'  # Destination folder for extraction
+
+unzip_and_save(zip_file_path, extraction_path)
+
 # Load the fine-tuned model and tokenizer
-model_path = "finetuned_model"
-tokenizer_path = "finetuned_tokenizer"
+model_path = "bert_model_sentiment_v1/finetuned_bert_sentiment_harsh"
+tokenizer_path = "bert_model_sentiment_v1/finetuned_bert_sentiment_harsh"
 
 @st.cache_resource
 def load_model():