update

app.py

@@ -4,8 +4,6 @@ import random
 import warnings
 
 import gradio as gr
-import librosa
-import numpy as np
 from datasets import IterableDatasetDict, load_dataset
 from gradio_client import Client
 from loguru import logger
@@ -13,7 +11,6 @@ from loguru import logger
 warnings.filterwarnings("ignore")
 
 NUM_TAR_FILES = 115
-NUM_SAMPLES = 3746131
 HF_PATH_TO_DATASET = "litagin/Galgame_Speech_SER_16kHz"
 
 hf_token = os.getenv("HF_TOKEN")
@@ -214,7 +211,7 @@ with gr.Blocks(head=shortcut_js) as app:
         logger.info("Putting unmodified")
         current_key = data[key]
         current_label_id = data[label_id]
-        _ = client.predict(
+        result = client.predict(
             new_data=json.dumps(
                 {
                     "key": current_key,
@@ -223,7 +220,7 @@ with gr.Blocks(head=shortcut_js) as app:
             ),
             api_name="/put_data",
         )
-        logger.info("Unmodified sent")
+        logger.info(f"Unmodified sent, result: {result}")
         return set_next_item(data)
 
     btn_init.click(
@@ -243,7 +240,7 @@ with gr.Blocks(head=shortcut_js) as app:
         def put_label(data: dict, _id=_id) -> dict:
             logger.info(f"Putting label: {id2rich_label[_id]}")
             current_key = data[key]
-            _ = client.predict(
+            result = client.predict(
                 new_data=json.dumps(
                     {
                         "key": current_key,
@@ -252,7 +249,7 @@ with gr.Blocks(head=shortcut_js) as app:
                 ),
                 api_name="/put_data",
             )
-            logger.info("Modified sent")
+            logger.info(f"Modified sent, result: {result}")
             return set_next_item(data)
 
         functions_list.append(put_label)

requirements.txt

@@ -1,5 +1,4 @@
 datasets
 gradio==4.44.1
-librosa
 loguru
 numpy<2.0