Update app.py

app.py

@@ -15,8 +15,8 @@ def predict(image: Image.Image, question: str) -> str:
 
     input_ids = None
     if question:
-        text = f"Please answer the question. Question: {question} Answer:" if '?' in question else question
-        input_ids = model.text_transform(text).input_ids.to(device)
+        prompt = f"Please answer the question. Question: {question} Answer:" if '?' in question else question
+        input_ids = model.text_transform(prompt).input_ids.to(device)
     
     generated_ids = model.generate(
         pixel_values=pixel_values,