Update app.py

app.py

@@ -11,8 +11,8 @@ parser = argparse.ArgumentParser()
 parser.add_argument("--cpu", action="store_true", help="Use CPU for computation")
 args = parser.parse_args([])
 
-DEVICE = "cuda" if torch.cuda.is_available() and not args.cpu else "cpu" # Determine device based on availability and argument
-DTYPE = torch.float32 if DEVICE == "cpu" else torch.float16 # CPU doesn't support float16
+DEVICE = "cpu"
+DTYPE = torch.float32 
 LATEST_REVISION = "2024-05-20"
 
 tokenizer = AutoTokenizer.from_pretrained("vikhyatk/moondream2", revision=LATEST_REVISION)