Update app.py

app.py

@@ -10,6 +10,7 @@ feature_extractor = ViTFeatureExtractor.from_pretrained('Dhahlan2000/freshness_d
 # Move to GPU if available
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = model.to(device)
+model.eval()
 
 # Class labels (modify according to your model)
 class_labels = [