wrong input

app.py

@@ -73,7 +73,7 @@ def text_to_image(image, prompt, duplications: float):
         batch['input_ids'].to(model.device)
         batch['attention_mask'].to(model.device)
         batch['pixel_values'].to(model.device)
-        output = model.generate(**inputs, max_new_tokens=500, temperature=0.3)
+        output = model.generate(**batch, max_new_tokens=500, temperature=0.3)
 
         # Unload GPU
         batch['input_ids'].to('cpu')