Update app.py

app.py

@@ -1,4 +1,4 @@
-REVISION = "4016e8293d864b9abfbc2a82a735c83f598d0b08"
+REVISION = "5ba9ed27c083d303bf67b461191cd6a45442ee47"
 
 try:
     import spaces
@@ -34,20 +34,15 @@ import json
 
 from queue import Queue
 from threading import Thread
-from transformers import (
-    TextIteratorStreamer,
-    AutoTokenizer,
-    AutoModelForCausalLM,
-)
+from transformers import AutoModelForCausalLM
 from PIL import ImageDraw
 from torchvision.transforms.v2 import Resize
 
 os.environ["HF_TOKEN"] = os.environ.get("TOKEN_FROM_SECRET") or True
-tokenizer = AutoTokenizer.from_pretrained("moondream/starmie-v1")
 moondream = AutoModelForCausalLM.from_pretrained(
     "vikhyatk/moondream-next",
     trust_remote_code=True,
-    torch_dtype=torch.float16,
+    torch_dtype=torch.bfloat16,
     device_map={"": "cuda"},
     revision=REVISION
 )