Update app.py

app.py

@@ -1,4 +1,4 @@
-REVISION = "faa59888f8e93586c753a353bdbf53d00213a7d1"
+REVISION = "9fd441468eb4ca6091a6fbee26edc1090b55ded7"
 
 try:
     import spaces
@@ -46,18 +46,6 @@ moondream = AutoModelForCausalLM.from_pretrained(
     device_map={"": "cuda"},
     revision=REVISION
 )
-
-# CKPT_DIRS = ["/tmp/md-ckpt/ckpt/ft/song-moon-4c-s15/s72001/"]
-# def get_ckpt(filename):
-#     ckpts = [
-#         torch.load(os.path.join(dir, filename), map_location="cpu") for dir in CKPT_DIRS
-#     ]
-#     avg_ckpt = {
-#         key.replace("._orig_mod", ""): sum(ckpt[key] for ckpt in ckpts) / len(ckpts)
-#         for key in ckpts[0]
-#     }
-#     return avg_ckpt
-# moondream.load_state_dict(get_ckpt("model.pt"))
 moondream.eval()
 
 
@@ -278,7 +266,7 @@ with gr.Blocks(title="moondream vl (new)", css=css, js=js) as demo:
                                 scale=4,
                             )
                             submit = gr.Button("Submit")
-                            reasoning = gr.Checkbox(label="Enable reasoning")
+                        reasoning = gr.Checkbox(label="Enable reasoning")
                         img = gr.Image(type="pil", label="Upload an Image")
                     submit.click(answer_question, [img, prompt, reasoning], [output, thought])
                     prompt.submit(answer_question, [img, prompt, reasoning], [output, thought])
@@ -362,16 +350,15 @@ with gr.Blocks(title="moondream vl (new)", css=css, js=js) as demo:
                 img = Resize(768)(img)
                 w, h = img.size
 
-            coords = json.loads(evt.value[1])
-            if len(coords) % 2 != 0:
-                raise ValueError("Only points supported right now.")
+            points = json.loads(evt.value[1])
 
             img_clone = img.copy()
             draw = ImageDraw.Draw(img_clone)
             
             for i in range(0, len(coords), 2):  # Step by 2 to handle x,y pairs
-                x = int(coords[i] * w)
-                y = int(coords[i + 1] * h)
+            for point in points:
+                x = int(point[0] * w)
+                y = int(point[1] * h)
                 draw.ellipse(
                     (x - 3, y - 3, x + 3, y + 3),
                     fill="red",