Update app.py

app.py

@@ -6,6 +6,12 @@ import gradio as gr
 MODEL_ID = "Qwen/Qwen2-VL-7B-Instruct"
 MODEL_FINETUNE_ID = "davidr99/qwen2.5-7b-instruct-blackjack"
 
+EXAMPLES = [
+    "examples/black_jack_screenshot_1737088587.png",
+    "examples/black_jack_screenshot_1737088629.png",
+    "examples/black_jack_screenshot_1737088648.png"
+]
+
 
 from transformers import Qwen2VLForConditionalGeneration, AutoTokenizer, AutoProcessor
 from qwen_vl_utils import process_vision_info
@@ -15,9 +21,9 @@ model.load_adapter(MODEL_FINETUNE_ID)
 processor = AutoProcessor.from_pretrained(MODEL_FINETUNE_ID)
 
 @spaces.GPU(duration=30)
-def blackjack_ai(image):
+def blackjack_ai(image, question):
 
-    instruction = "extract json from this image."
+    instruction = question
     
     messages = [
         {"role": "system",
@@ -60,9 +66,11 @@ def blackjack_ai(image):
 with gr.Blocks() as demo:
    
     image = gr.Image(type="filepath")
+    question = gr.Textbox(value = "extract json from this image.")
     submit = gr.Button("Submit")
     output = gr.TextArea()
+    examples = gr.Examples(examples=EXAMPLES, inputs=[image])
 
-    submit.click(blackjack_ai, inputs=[image], outputs=[output])
+    submit.click(blackjack_ai, inputs=[image, question], outputs=[output])
     
 demo.launch()