Update app.py

app.py

@@ -20,7 +20,7 @@ from transformers import AutoProcessor
 model_name = "ahmed-masry/ColFlor"
 token = os.environ.get("HF_TOKEN")
 model = ColFlor.from_pretrained(
-    model_name, torch_dtype=torch.bfloat16, device_map="cuda", token = token).eval()
+    model_name, device_map="cuda", token = token).eval()
 
 processor = ColFlorProcessor.from_pretrained(model_name, token = token)