Update app.py

app.py

@@ -84,7 +84,7 @@ chat_history = []
 
 # Function to handle interaction with model
 @spaces.GPU
-def interact(user_input, history, interaction_count, model_name):
+def interact(user_input, history, interaction_count):
     global tokenizer, model
     try:
         if tokenizer is None or model is None:
@@ -151,7 +151,7 @@ Here is the story:
                 chat_history = []  # Reset chat history
                 chat_history.append({"role": "system", "content": combined_message})
                 question_prompt = "Please ask a simple question about the story to encourage interaction."
-                _, formatted_history, chat_history, interaction_count = interact(question_prompt, chat_history, interaction_count, model_name)
+                _, formatted_history, chat_history, interaction_count = interact(question_prompt, chat_history, interaction_count)
 
                 return formatted_history, chat_history, gr.update(value=[]), story["story"]
             else: