Update app.py

app.py

@@ -1,4 +1,4 @@
-REVISION = "cbf0882aba4a68e10752fb2f9d853fd9b5df5ec9"
+REVISION = "e260c8a27acb2bb5fcc64537bb1b52b588dd575a"
 
 try:
     import spaces
@@ -145,17 +145,20 @@ def caption(img, mode):
         yield buffer.strip()
 
 @spaces.GPU(duration=10)
-def detect(img, object):
+def detect(img, object, eos_bias):
     if img is None:
         yield "", gr.update(visible=False, value=None)
         return
 
+    eos_bias = float(eos_bias)
+
+    objs = moondream.detect(img, object, settings={"eos_bias": eos_bias})["objects"]
+    
     w, h = img.size
     if w > 768 or h > 768:
         img = Resize(768)(img)
         w, h = img.size
 
-    objs = moondream.detect(img, object)["objects"]
     draw_image = ImageDraw.Draw(img)
     for o in objs:
         draw_image.rectangle(
@@ -295,9 +298,10 @@ with gr.Blocks(title="moondream vl (new)", css=css, js=js) as demo:
                             )
                             submit = gr.Button("Submit")
                         img = gr.Image(type="pil", label="Upload an Image")
-                    submit.click(detect, [img, prompt], [thought, ann])
-                    prompt.submit(detect, [img, prompt], [thought, ann])
-                    img.change(detect, [img, prompt], [thought, ann])
+                        eos_bias = gr.Textbox(label="EOS Bias", value="0")
+                    submit.click(detect, [img, prompt, eos_bias], [thought, ann])
+                    prompt.submit(detect, [img, prompt, eos_bias], [thought, ann])
+                    img.change(detect, [img, prompt, eos_bias], [thought, ann])
                 elif mode == "Point":
                     with gr.Group():
                         with gr.Row():