Spaces:

andresgtn
/

bean-leaf-health-classifier

Runtime error

andresgtn commited on Sep 23, 2022

Commit

f6f1dea

•

1 Parent(s): ada2f3f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import numpy as np
 import gradio as gr
 import torch
-dataset = load_dataset("beans") # This should be the same as the first line of Python code in this Colab notebook
 extractor = AutoFeatureExtractor.from_pretrained("andresgtn/vit-base-bean-health-classifier")
 model = AutoModelForImageClassification.from_pretrained("andresgtn/vit-base-bean-health-classifier")

 import gradio as gr
 import torch
+dataset = datasets.load_dataset("beans") # This should be the same as the first line of Python code in this Colab notebook
 extractor = AutoFeatureExtractor.from_pretrained("andresgtn/vit-base-bean-health-classifier")
 model = AutoModelForImageClassification.from_pretrained("andresgtn/vit-base-bean-health-classifier")