Update app.py

app.py

@@ -1,3 +1,4 @@
+import warnings
 import numpy as np
 import pandas as pd
 import os
@@ -13,6 +14,9 @@ from transformers import GPTJForCausalLM, GPT2TokenizerFast
 import torch
 import torch.multiprocessing as mp
 
+# Suppress specific warnings
+warnings.filterwarnings('ignore', category=FutureWarning, module='huggingface_hub.file_download')
+
 # Initialize Example Emotions Dataset
 data = {
     'context': [
@@ -207,8 +211,8 @@ def get_emotional_response(context):
 
 # Initialize the pre-trained language model
 model_name = 'EleutherAI/gpt-j-6B'
-tokenizer = GPT2TokenizerFast.from_pretrained(model_name)
-lm_model = GPTJForCausalLM.from_pretrained(model_name)
+tokenizer = GPT2TokenizerFast.from_pretrained(model_name, force_download=True)
+lm_model = GPTJForCausalLM.from_pretrained(model_name, force_download=True)
 
 # Multiprocessing context setting (ensure it's set only once)
 if __name__ == '__main__':