Upload app.py

app.py

@@ -45,15 +45,14 @@ streamer = TextIteratorStreamer(tokenizer, timeout=10.0, skip_prompt=True, skip_
 
 @spaces.GPU(duration=59)
 def chat(message: str, 
-         history: list, 
+         history: list[dict], 
          temperature: float, 
          max_new_tokens: int,
          progress=gr.Progress(track_tqdm=True)
         ):
     try:
-        conversation = [{"role": "user", "content": message}]
-        if len(history) == 0: history = conversation
-        else: history.extend(conversation)
+        if not history: history = []
+        history.append({"role": "user", "content": message})
 
         input_tensors = tokenizer.apply_chat_template(history, add_generation_prompt=True, return_dict=True, return_tensors="pt").to(model.device)