File size: 562 Bytes
c2cbc53
2d931f2
8142bba
2d931f2
8142bba
2d931f2
8142bba
2d931f2
fc7c04c
a425649
8142bba
a425649
2d931f2
8142bba
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import gradio as gr
from transformers import AutoFeatureExtractor, AutoModelForImageClassification, pipeline

extractor = AutoFeatureExtractor.from_pretrained("edixo/road_good_damaged_condition")

model = AutoModelForImageClassification.from_pretrained("edixo/road_good_damaged_condition")

pipe = pipeline(task="image-classification", 
                model=model,
                feature_extractor=extractor)

gr.Interface.from_pipeline(pipe, 
                    title="roads",
                    description="Add description",
                    ).launch()