yusufbardolia
/

phase_2b

Model card Files Files and versions

yusufbardolia commited on Jan 15

Commit

25b8858

·

verified ·

1 Parent(s): 7c50cf5

Update script.py

Files changed (1) hide show

script.py +3 -10

script.py CHANGED Viewed

@@ -94,16 +94,9 @@ if __name__ == "__main__":
     model.to(device).eval()
     PROMPT = (
-        "Monopolar Curved Scissors. "
-        "curved surgical scissors. "
-        "surgical instrument with blades. "
-        "Prograsp Forceps. "
-        "surgical forceps grasping tissue. "
-        "metal grasping instrument. "
-        "Large Needle Driver. "
-        "needle holder surgical instrument. "
-        "laparoscopic surgical instrument. "
-        "robotic surgical instrument."
     )
     BOX_THRESHOLD = 0.28

     model.to(device).eval()
     PROMPT = (
+        "Monopolar Curved Scissors . surgical scissors cutting . "
+        "Prograsp Forceps . robotic grasper jaws . "
+        "Large Needle Driver . needle holder . "
     )
     BOX_THRESHOLD = 0.28