Update app.py

app.py

@@ -8,6 +8,10 @@ from datasets import load_dataset
 from transformers import TrainingArguments, ViTConfig, ViTForImageClassification
 from torchvision import transforms
 import pandas as pd
+import logging
+
+logging.basicConfig(level=logging.debug)
+logger = logging.getLogger(__name__)
 
 def load_config(config_path):
     with open(config_path, 'r') as f:
@@ -131,7 +135,7 @@ def main():
                 output = gr.JSON(label="Classification Result")
         input_image.change(fn=simple_predict, inputs=input_image, outputs=output)
     # Launch the Gradio app
-    demo.launch()
+    demo.launch(show_error=True)
 
 if __name__ == "__main__":
     main()