Feature upload a whole folder

app.py

@@ -3,15 +3,24 @@ import numpy as np
 import pandas as pd
 import os
 from tkinter import Tk, filedialog
-
 import gradio as gr
 from pathlib import Path
+from transformers import AutoImageProcessor, AutoModelForImageClassification
+import torch
+
+# Load model
+processor = AutoImageProcessor.from_pretrained("victor/animals-classifier")
+model = AutoModelForImageClassification.from_pretrained("victor/animals-classifier")
+model.eval()
 
 
 def upload_file(files):
-    file_paths = [file.name for file in files]
-    df_file_paths = pd.DataFrame(file_paths, columns=["file_name"])
-    return df_file_paths
+    dir_list = [file.name for file in files]
+    df = pd.DataFrame(columns=["file_name", "label", "accuracy"])
+    df["file_name"] = dir_list
+    df["label"] = "incoming"
+    df["accuracy"] = 100
+    return df
 
 
 with gr.Blocks() as demo:

requirement.txt

@@ -20,7 +20,21 @@ Jinja2==3.1.5
 markdown-it-py==3.0.0
 MarkupSafe==2.1.5
 mdurl==0.1.2
+mpmath==1.3.0
+networkx==3.4.2
 numpy==2.2.2
+nvidia-cublas-cu12==12.4.5.8
+nvidia-cuda-cupti-cu12==12.4.127
+nvidia-cuda-nvrtc-cu12==12.4.127
+nvidia-cuda-runtime-cu12==12.4.127
+nvidia-cudnn-cu12==9.1.0.70
+nvidia-cufft-cu12==11.2.1.3
+nvidia-curand-cu12==10.3.5.147
+nvidia-cusolver-cu12==11.6.1.9
+nvidia-cusparse-cu12==12.3.1.170
+nvidia-nccl-cu12==2.21.5
+nvidia-nvjitlink-cu12==12.4.127
+nvidia-nvtx-cu12==12.4.127
 orjson==3.10.15
 packaging==24.2
 pandas==2.2.3
@@ -33,18 +47,24 @@ python-dateutil==2.9.0.post0
 python-multipart==0.0.20
 pytz==2024.2
 PyYAML==6.0.2
+regex==2024.11.6
 requests==2.32.3
 rich==13.9.4
 ruff==0.9.3
 safehttpx==0.1.6
+safetensors==0.5.2
 semantic-version==2.10.0
 setuptools==75.1.0
 shellingham==1.5.4
 six==1.17.0
 sniffio==1.3.1
 starlette==0.45.3
+sympy==1.13.1
+tokenizers==0.21.0
 tomlkit==0.13.2
+torch==2.5.1
 tqdm==4.67.1
+transformers==4.48.1
 typer==0.15.1
 typing_extensions==4.12.2
 tzdata==2025.1