chore: upgrade space to gradio 6

README.md

@@ -4,7 +4,8 @@ emoji: 💡
 colorFrom: yellow
 colorTo: gray
 sdk: gradio
-sdk_version: 4.37.1
+sdk_version: 6.14.0
+python_version: "3.11"
 app_file: app.py
 pinned: false
 license: mit
@@ -12,4 +13,4 @@ license: mit
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
-["Automatic Assessment of Divergent Thinking in Chinese Language with TransDis: A Transformer-Based Language Model Approach"](https://arxiv.org/abs/2306.14790)
+["Automatic Assessment of Divergent Thinking in Chinese Language with TransDis: A Transformer-Based Language Model Approach"](https://arxiv.org/abs/2306.14790)

app.py

@@ -1,3 +1,5 @@
+import os
+import tempfile
 import traceback
 from io import StringIO
 from typing import Optional
@@ -10,7 +12,18 @@ from utils import pipeline
 from utils.models import list_models
 
 
+def resolve_file_path(file) -> str:
+    if isinstance(file, dict) and file.get('path'):
+        return os.fspath(file['path'])
+    if isinstance(file, (str, os.PathLike)):
+        return os.fspath(file)
+    if hasattr(file, 'name'):
+        return os.fspath(file.name)
+    raise TypeError(f'Unsupported file input: {type(file)!r}')
+
+
 def read_data(filepath: str) -> Optional[pd.DataFrame]:
+    filepath = os.fspath(filepath)
     if filepath.endswith('.xlsx'):
         df = pd.read_excel(filepath)
     elif filepath.endswith('.csv'):
@@ -31,7 +44,7 @@ def process(
         logger.info(f'Processing {task_name} with {model_name} and {pooling}')
         # load file
         if file:
-            df = read_data(file.name)
+            df = read_data(resolve_file_path(file))
         elif text:
             string_io = StringIO(text)
             df = pd.read_csv(string_io)
@@ -52,11 +65,12 @@ def process(
             raise Exception('Task not supported')
 
         # save
-        path = 'output.csv'
+        fd, path = tempfile.mkstemp(prefix='transdis_', suffix='.csv')
+        os.close(fd)
         df.to_csv(path, index=False, encoding='utf-8-sig')
         return None, df.iloc[:10], path
 
-    except:
+    except Exception:
         error = traceback.format_exc()
         logger.warning({
             'error': error,
@@ -66,7 +80,7 @@ def process(
             'text': text,
             'file': file,
         })
-        return {'Info': 'Something wrong', 'Error': traceback.format_exc()}, None, None
+        return f'Something wrong\n\n{error}', None, None
 
 
 # input
@@ -104,4 +118,5 @@ app = gr.Interface(
     title='TransDis-CreativityAutoAssessment',
     concurrency_limit=1,
 )
-app.launch(max_threads=1)
+if __name__ == '__main__':
+    app.launch(max_threads=1)

requirements.txt

@@ -1,12 +1,12 @@
-numpy
-pandas
+numpy<3
+pandas>=2.2,<4
 
-transformers
-sentence-transformers
+transformers>=4.45,<6
+sentence-transformers>=3,<6
 
-torch
+torch>=2.4,<3
 
-openpyxl
-tabulate
-loguru
+openpyxl>=3.1,<4
+tabulate>=0.9,<1
+loguru>=0.7,<1
 # gradio