Update app.py

app.py

@@ -1,6 +1,6 @@
 import os
+import zipfile
 import numpy as np
-import gradio as gr
 import matplotlib.pyplot as plt
 import tensorflow as tf
 from tensorflow.keras.preprocessing.image import ImageDataGenerator
@@ -9,10 +9,15 @@ from tensorflow.keras.layers import Dense, GlobalAveragePooling2D, Dropout, Batc
 from tensorflow.keras.models import Model
 from tensorflow.keras.optimizers import Adam
 
-# Suppress TensorFlow logging and warnings
-os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'  # Suppress TensorFlow logging (1 = INFO, 2 = WARNING, 3 = ERROR)
-tf.get_logger().setLevel('ERROR')
-tf.autograph.set_verbosity(2)
+# Function to download and extract the dataset
+def download_and_extract_dataset(dataset_name):
+    # Make sure the kaggle.json file is in the proper location
+    os.environ['KAGGLE_CONFIG_DIR'] = os.path.expanduser("~/.kaggle/")
+    # Download the dataset
+    os.system(f'kaggle datasets download -d mikoajkolman/pokemon-images-first-generation17000-files')
+    # Extract the dataset
+    with zipfile.ZipFile(f'{dataset_name}.zip', 'r') as zip_ref:
+        zip_ref.extractall('.')
 
 # Function to find the greatest batch size that fully divides the total samples
 def find_batch_size(total_samples, initial_batch_size):
@@ -21,10 +26,14 @@ def find_batch_size(total_samples, initial_batch_size):
             return bs
     return 1  # Fallback to 1 if no perfect divisor is found
 
-# Path to the image data directory
-base_dir = 'pokemon-images-first-generation17000-files/pokemon'  # Adjust path as needed
-if not os.path.exists(base_dir):
-    raise Exception("The specified base path does not exist. Please verify the path.")
+# Suppress TensorFlow logging and warnings
+os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'  # Suppress TensorFlow logging (1 = INFO, 2 = WARNING)
+tf.get_logger().setLevel('ERROR')
+tf.autograph.set_verbosity(2)
+
+# Download and extract the dataset
+dataset_name = 'mikoajkolman/pokemon-images-first
+
 
 # Define the Pokémon classes to be classified
 classes = ['Doduo', 'Geodude', 'Zubat']