Upload app.py

app.py

@@ -12,7 +12,8 @@ import subprocess
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-model_id = "vikhyatk/moondream2"
+#model_id = "vikhyatk/moondream2"
+model_id = "Xenova/moondream2"
 #model_id = "zesquirrelnator/moondream2-finetuneV2"
 #revision = "2024-08-26"
 #tokenizer = AutoTokenizer.from_pretrained(model_id, revision=revision)
@@ -25,7 +26,6 @@ tokenizer = AutoTokenizer.from_pretrained(model_id)
 #).to(device)
 moondream = ORTModelForImageClassification.from_pretrained(
     model_id, trust_remote_code=True,
-    from_transformers=True
 ).to(device)
 
 moondream.eval()