Update app.py

app.py

@@ -1,5 +1,26 @@
 import os
 import zipfile
+
+zip_file = "Fruits_datasets.zip"
+dataset_folder = "./Data"
+
+# Unzip only if not already done
+if not os.path.exists(dataset_folder):
+    print("Extracting dataset from uploaded zip...")
+    with zipfile.ZipFile(zip_file, 'r') as zip_ref:
+        zip_ref.extractall(".")
+else:
+    print("Dataset already extracted.")
+
+# Now you can access:
+for cls in os.listdir(dataset_folder):
+    print("Class:", cls)
+    cls_path = os.path.join(dataset_folder, cls)
+    print("Files:", os.listdir(cls_path))
+
+"""
+import os
+import zipfile
 import torch
 import torch.nn as nn
 import torch.optim as optim
@@ -197,6 +218,6 @@ with desc_block:
         image_input = gr.Image(type="pil", label="Upload Fruit Image", image_mode="RGB", height=256, width=256)
         pred_output = gr.Textbox(label="Prediction Result")
         predict_button = gr.Button("Classify Image")
-        predict_button.click(fn=predict_image, inputs=image_input, outputs=pred_output)
+        predict_button.click(fn=predict_image, inputs=image_input, outputs=pred_output)"""
 
 desc_block.launch()