Spaces:
Runtime error
Runtime error
import transformers | |
from transformers import ViTModel, ViTImageProcessor | |
from torchvision import transforms | |
import gradio as gr | |
model = ViTModel.from_pretrained("afern24/vit-beans-finetuned") | |
def classify_image(inp): | |
inp = transforms.ToTensor()(inp).unsqueeze(0) | |
processor = ViTImageProcessor.from_pretrained('google/vit-base-patch16-224-in21k') | |
inputs = processor(images=inp, return_tensors="pt") | |
outputs = model(**inputs) | |
return outputs | |
# Define your examples as a list of lists | |
examples = [["leaf1.jpg"], ["leaf2.jpg"], ["leaf3.jpg"]] | |
# Load your interface with examples | |
gr.Interface(fn=classify_image, inputs=gr.inputs.Image(type="pil"), outputs=gr.outputs.Label(num_top_classes=3), examples=examples).launch(debug=True) | |