Update app.py

app.py

@@ -20,7 +20,7 @@ question = st.text_input("Question:", "")
 
 # Prepare the input text
 input_text = f"question: {question} context: {context}"
-input_ids = tokenizer.encode(input_text, return_tensors="pt").to(device)
+input_ids = tokenizer.encode(input_text, return_tensors="pt")
 
 # Generate the answer