Update app.py

app.py

@@ -1,4 +1,4 @@
-REVISION = "02ddf9ce6f292965305f578ab7d070e59b24d0b1"
+REVISION = "c2bcd8ac7d7dabbbd84f0176f6886395577625de"
 
 try:
     import spaces
@@ -149,7 +149,13 @@ def caption(img, mode):
         return
 
     buffer = ""
-    for t in moondream.caption(img, length="short" if mode == "Short" else "normal", stream=True)["caption"]:
+    if mode == "Short":
+        l = "short"
+    elif mode == "Long":
+        l = "long"
+    else:
+        l = "normal"
+    for t in moondream.caption(img, length=l, stream=True)["caption"]:
         buffer += t
         yield buffer.strip()