Fix function

app.py

@@ -12,8 +12,8 @@ tokenizer = AutoTokenizer.from_pretrained("gsarti/phi3-mini-rebus-solver-fp16")
 model = AutoModelForCausalLM.from_pretrained("gsarti/phi3-mini-rebus-solver-fp16")
 
 @spaces.GPU
-def solve_verbalized_rebus(ex):
-    input = template.format(input=ex)
+def solve_verbalized_rebus(example, history):
+    input = template.format(input=example)
     inputs = tokenizer(input, return_tensors="pt")["input_ids"]
     outputs = model.generate(input_ids = inputs, max_new_tokens = 500, use_cache = True)
     model_generations = tokenizer.batch_decode(outputs)