Update app.py

app.py

@@ -5,8 +5,8 @@ import gradio as gr
 
 dataset = load_dataset("beans") # This should be the same as the first line of Python code in this Colab notebook
 
-extractor = AutoFeatureExtractor.from_pretrained("saved_model_files")
-model = AutoModelForImageClassification.from_pretrained("saved_model_files")
+extractor = AutoFeatureExtractor.from_pretrained("andresgtn/vit-base-bean-health-classifier")
+model = AutoModelForImageClassification.from_pretrained("andresgtn/vit-base-bean-health-classifier")
 
 # add to cuda?
 model.eval()