Update app.py

app.py

@@ -64,7 +64,7 @@ emotion_prediction_tokenizer = AutoTokenizer.from_pretrained("bhadresh-savani/di
 response_model_name = "gpt2-xl"
 response_tokenizer = AutoTokenizer.from_pretrained(response_model_name)
 with init_empty_weights():
-    response_model = AutoModelForCausalLM.from_pretrained(response_model_name, device_map="auto", torch_dtype=torch.float16, assign=True)
+    response_model = AutoModelForCausalLM.from_pretrained(response_model_name, device_map="auto", torch_dtype=torch.float16)
 
 # Set the pad token
 response_tokenizer.pad_token = response_tokenizer.eos_token