Update app.py

app.py

@@ -6,6 +6,9 @@ Automatically generated by Colaboratory.
 Original file is located at
     https://colab.research.google.com/drive/1XX8pCT291obpzL4fc1vu5L_HTG027lle
 """
+import gradio
+import datasets
+from transformers import AutoFeatureExtractor, AutoModelForImageClassification
 
 dataset = datasets.load_dataset("beans") # This should be the same as the first line of Python code in this Colab notebook