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