Fix bug in gradio

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
zilliztech · Oct 25, 2023 · 2e2486e · 2e2486e
1 parent 936c6b2
commit 2e2486e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/gradio_demo.py b/gradio_demo.py
@@ -44,9 +44,10 @@ def clear_memory(project, session):
 
 def add_project(project, data_url: str = None, data_file: object = None):
     if data_file:
-        return insert(data_src=data_file.name, project=project, source_type='file')
+        num_chunks, _ = insert(data_src=data_file.name, project=project, source_type='file')
     if data_url:
-        return insert(data_src=data_url, project=project, source_type='url')
+        num_chunks, _ = insert(data_src=data_url, project=project, source_type='url')
+    return num_chunks
 
 
 def check_project(project):