Update app.py

app.py

@@ -47,7 +47,7 @@ terminators = [
     tokenizer.convert_tokens_to_ids("")
 ]
 
-def chat_llama3_8b(message: str, history: list, temperature: float, max_new_tokens: int) -> str:
+def chat_llama3_8b(message: str, history: list, temperature: float, max_new_tokens: int):
     conversation = []
     for user, assistant in history:
         conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
@@ -58,7 +58,7 @@ def chat_llama3_8b(message: str, history: list, temperature: float, max_new_toke
     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,