Cleaned Up App and Added Requirements

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
-from transformers import AutoFeatureExtractor, AutoModelForImageClassification, AutoTokenizer, AutoModelForSequenceClassification
-from transformers import pipeline 
+from transformers import AutoFeatureExtractor, AutoModelForImageClassification, pipeline 
 
 image_model = AutoModelForImageClassification.from_pretrained("google/vit-base-patch16-224")
 extractor = AutoFeatureExtractor.from_pretrained("google/vit-base-patch16-224")

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+pytorch
+transformers