Avmromanov commited on
Commit
9a12af7
·
1 Parent(s): d9c4377
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
  # Load a model trained on car brands/makes
5
  # Using a model from the Hugging Face hub that's good at vehicle classification
6
  car_classifier = pipeline("image-classification",
7
- model="dima806/car_brand_classification")
8
 
9
  # Alternative models you can try:
10
  # - "nickmuchi/vit-finetuned-cars-biased"
 
4
  # Load a model trained on car brands/makes
5
  # Using a model from the Hugging Face hub that's good at vehicle classification
6
  car_classifier = pipeline("image-classification",
7
+ model="dima806/car_models_image_detection")
8
 
9
  # Alternative models you can try:
10
  # - "nickmuchi/vit-finetuned-cars-biased"