Update app.py

app.py

@@ -52,7 +52,7 @@ def stream_chat(message: str, history: list, temperature: float, max_new_tokens:
     print(f"Conversation is -\n{conversation}")
     
     input_ids = tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)
-    inputs = tokenizer(input_ids, return_tensors="pt").to(0) #gpu 0, cpu 1
+    inputs = tokenizer(input_ids, return_tensors="pt").to('cpu') #gpu 0, cpu 1
     
     streamer = TextIteratorStreamer(tokenizer, timeout=60., skip_prompt=True, skip_special_tokens=True)