Update app.py

app.py

@@ -5,14 +5,15 @@ import torch
 from torchvision.io import read_image
 from transformers import ViTForImageClassification, ViTFeatureExtractor,ViTImageProcessor,pipeline
 
+# With ViTImageProcessor we can't use pipeline for image classification so Icomment it
 # model = ViTImageProcessor.from_pretrained('SeyedAli/Food-Image-Classification-VIT')
+
 model = ViTForImageClassification.from_pretrained('SeyedAli/Food-Image-Classification-VIT')
 feature_extractor = ViTFeatureExtractor.from_pretrained('SeyedAli/Food-Image-Classification-VIT')
+
 def FoodClassification(image):
     with tempfile.NamedTemporaryFile(suffix=".png") as temp_image_file:
         # Copy the contents of the uploaded image file to the temporary file
-        # temp_image_file.write(open(image, "rb").read())
-        # temp_image_file.flush()
         Image.fromarray(image).save(temp_image_file.name)
         # Load the image file using torchvision
         image = read_image(temp_image_file.name)