Update app.py

app.py

@@ -70,7 +70,6 @@ def process_text(text):
     return text
 
 
-# Gradio Function
 @spaces.GPU
 def generate(
     model_choice: str,
@@ -89,9 +88,17 @@ def generate(
     else:
         model = editing_model
         tokenizer = editing_tokenizer
-        
-    for user, assistant in chat_history:
-        conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
+
+    # Checking each tuple in chat_history to ensure it has exactly two elements
+    for item in chat_history:
+        if isinstance(item, tuple) and len(item) == 2:
+            user, assistant = item
+            conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
+        else:
+            print(f"Error in chat history item: {item}. Each item must be a tuple with exactly two elements.")
+            continue  # Skip this item or handle appropriately
+
+    # Append the current user message
     conversation.append({"role": "user", "content": message})
 
     input_ids = tokenizer.apply_chat_template(conversation, return_tensors="pt")
@@ -102,7 +109,7 @@ def generate(
 
     streamer = TextIteratorStreamer(tokenizer, timeout=10.0, skip_prompt=True, skip_special_tokens=True)
     generate_kwargs = dict(
-        {"input_ids": input_ids},
+        input_ids=input_ids,
         streamer=streamer,
         max_new_tokens=max_new_tokens,
         do_sample=True,