Update app.py

app.py

@@ -59,7 +59,7 @@ def load_model(model_name):
         tokenizer = AutoTokenizer.from_pretrained(
             model_path, 
             padding_side='left', 
-            token=hugging_face_token, 
+            use_auth_token=hugging_face_token, 
             trust_remote_code=True
         )
         
@@ -70,7 +70,7 @@ def load_model(model_name):
         
         model = AutoModelForCausalLM.from_pretrained(
             model_path, 
-            token=hugging_face_token, 
+            use_auth_token=hugging_face_token, 
             trust_remote_code=True
         )
         
@@ -85,6 +85,7 @@ def load_model(model_name):
     return tokenizer, model
 
 
+
 # Ensure the initial model is loaded
 tokenizer, model = load_model(selected_model)
 
@@ -146,7 +147,7 @@ def send_selected_story(title, model_name, system_prompt):
     global data  # Ensure data is reset
     data = []  # Reset data for new story
     interaction_count = 1  # Reset interaction count for new story
-    tokenizer, model = load_model(model_name)
+    tokenizer, model = load_model(model_name)  # Load the appropriate model
     selected_story = title
     for story in stories:
         if story["title"] == title:
@@ -170,6 +171,7 @@ Here is the story:
         else:
             print("Story title does not match.")
 
+
 # Function to save comment and score
 def save_comment_score(chat_responses, score, comment, story_name, user_name, system_prompt):
     full_chat_history = ""