Spaces:

tusharagarwal3
/

Tushar-image_classification_corise

App Files Files Community

tusharagarwal3 commited on Sep 25, 2022

Commit

c452031

•

1 Parent(s): b608dfd

Update app.py

Files changed (1) hide show

app.py +1 -1

app.py CHANGED Viewed

@@ -1,6 +1,6 @@
 from datasets import load_dataset
 from transformers import AutoFeatureExtractor, AutoModelForImageClassification
 # This should be the same as the first line of Python code in this Colab notebook
 dataset = load_dataset('beans')
 feature_extractor = AutoFeatureExtractor.from_pretrained("saved_model_files")

 from datasets import load_dataset
 from transformers import AutoFeatureExtractor, AutoModelForImageClassification
+import torch
 # This should be the same as the first line of Python code in this Colab notebook
 dataset = load_dataset('beans')
 feature_extractor = AutoFeatureExtractor.from_pretrained("saved_model_files")