Update utils.py

utils.py

@@ -5,12 +5,16 @@ from PIL import Image
 import io
 
 
-def answer(input_img):
+def answer(input_img, selected):
     buffer = io.BytesIO()
     input_img.save(buffer, format="PNG")
     image_data = base64.b64encode(buffer.getvalue()).decode('utf-8')
-    # print(os.getenv('PROMPT'))
 
-    respose = executor.execute_with_image(user_input= '', model_input= os.environ['PROMPT'], base64_image=image_data)
+    if selected == "Describe things":
+        model_input = os.environ['PROMPT']
+    else:
+        model_input = os.environ['PROMPT_OCR']
+
+    respose = executor.execute_with_image(user_input= '', model_input= model_input, base64_image=image_data)
 
     return respose