Update app.py

app.py

@@ -69,12 +69,12 @@ def chat_llama3_8b(message: str,
         conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
     conversation.append({"role": "user", "content": message})
 
-    input_ids = tokenizer.apply_chat_template(conversation, return_tensors="pt").to(model.device)
+    input_ids = tokenizer.apply_chat_template(conversation, return_tensors="pt")
     
     streamer = TextIteratorStreamer(tokenizer, timeout=10.0, skip_prompt=True, skip_special_tokens=True)
 
     generate_kwargs = dict(
-        input_ids= input_ids,
+        input_ids= input_ids.to(model.device),
         streamer=streamer,
         max_new_tokens=max_new_tokens,
         do_sample=True,