Spaces:

Kavindu99
/

leaf-classifier

Runtime error

Kavindu99 commited on Sep 25, 2022

Commit

d7de6c9

1 Parent(s): c1d35be

Updates to app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,8 +4,8 @@ import gradio as gr
 dataset = datasets.load_dataset("beans")
-extractor = AutoFeatureExtractor.from_pretrained("Kavindu99/leaf-classifier")
-model = AutoModelForImageClassification.from_pretrained("Kavindu99/leaf-classifier")
 labels = dataset['train'].features['labels'].names

 dataset = datasets.load_dataset("beans")
+extractor = AutoFeatureExtractor.from_pretrained("saved_model_files")
+model = AutoModelForImageClassification.from_pretrained("saved_model_files")
 labels = dataset['train'].features['labels'].names