Fix bug in gradio (#87)

Signed-off-by: Jael Gu <[email protected]>
zilliztech · Oct 25, 2023 · 37db7d6 · 37db7d6
1 parent 936c6b2
commit 37db7d6
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):