swarms/playground/models/fast_vit_example.py

6 lines
133 B

from swarms.models.fastvit import FastViT
fastvit = FastViT()
result = fastvit(img="images/swarms.jpeg", confidence_threshold=0.5)