Update app.py

app.py

@@ -169,24 +169,11 @@ def caption(img, mode):
         yield ""
         return
 
-    streamer = TextIteratorStreamer(tokenizer, skip_special_tokens=True)
-    thread = Thread(
-        target=moondream.caption,
-        kwargs={
-            "images": [img],
-            "length": "short" if mode == "Short" else None,
-            "tokenizer": tokenizer,
-            "streamer": streamer,
-        },
-    )
-    thread.start()
-
     buffer = ""
-    for new_text in streamer:
-        buffer += new_text
+    for t in moondream.caption(length="short" if mode == "Short" else None, stream=True)["caption"]:
+        buffer += t
         yield buffer.strip()
 
-
 @spaces.GPU(duration=10)
 def detect(img, object):
     if img is None: