arpm01's picture
Update app.py
2d931f2
raw
history blame
562 Bytes
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()