Spaces:
Runtime error
Runtime error
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() |