francis-ogbuagu
/

maize_vit_model

Image Classification

Maize Disease Detection

Model card Files Files and versions

francis-ogbuagu commited on 13 days ago

Commit

af494a7

·

verified ·

1 Parent(s): 93fde2d

Update app.py

Files changed (1) hide show

app.py +1 -1

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from transformers import ViTForImageClassification, AutoImageProcessor
 # Load model & processor
-model_path = "maize_vit_model"
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = ViTForImageClassification.from_pretrained(model_path, use_safetensors=True)

 # Load model & processor
+model_path = "model"
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = ViTForImageClassification.from_pretrained(model_path, use_safetensors=True)