Update app.py

app.py

@@ -161,7 +161,7 @@ def resume_QA(message, history):
     add_generation_prompt=True
     )   
 
-    model_inputs = tokenizer([text], return_tensors="pt").to("cuda")
+    model_inputs = tokenizer([text], return_tensors="pt")
     streamer = TextIteratorStreamer(tokenizer, timeout=50., skip_prompt=True, skip_special_tokens=True)
     generate_kwargs = dict(
         model_inputs,