Fix: input type error

app.py

@@ -5,8 +5,6 @@ from src.brain import generate_answers
 
 def response(query, history):
     global processing
-    if not pdf_loaded or processing:
-        return "Please wait...", history
     processing = True
     output = generate_answers(query)
     history.append((query, output))

src/brain.py

@@ -6,6 +6,6 @@ model = AutoModelForCausalLM.from_pretrained(model_name)
 
 
 def generate_answers(query):
-    input_ids = tokenizer(**query, return_tensors="pt")
-    output = model.generate(input_ids)
+    input_ids = tokenizer(query, return_tensors="pt")
+    output = model.generate(**input_ids)
     return tokenizer.decode(output[0])