fix

neoguojing · Aug 11, 2024 · 2a7724d · 2a7724d
1 parent f330191
commit 2a7724d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/Makefile b/Makefile
@@ -30,10 +30,11 @@ push:
 
 # Define the run command
 run:
+	@echo $(HAS_CUDA)
 	@if [ "$(HAS_CUDA)" -eq "1" ]; then \
-		docker run --gpus all -p 7860:7860 -v $(HOME)/.cache/huggingface:$(HOME)/.cache/huggingface -v $(HOME)/.deepface:$(HOME)/.deepface --rm -it --name ai-world $(IMAGE_NAME):$(IMAGE_TAG)-$(GIT_COMMIT); \
+		docker run --gpus all -p 7860:7860 -v $(HOME)/.cache/huggingface:/root/.cache/huggingface -v $(HOME)/.deepface:/root/.deepface --rm -it --name ai-world $(IMAGE_NAME):$(IMAGE_TAG)-$(GIT_COMMIT); \
 	else \
-		docker run -p 7860:7860 -v $(HOME)/.cache/huggingface:$(HOME)/.cache/huggingface -v $(HOME)/.deepface:$(HOME)/.deepface --rm -it --name ai-world $(IMAGE_NAME):$(IMAGE_TAG)-$(GIT_COMMIT); \
+		docker run -p 7860:7860 -v $(HOME)/.cache/huggingface:/root/.cache/huggingface -v $(HOME)/.deepface:/root/.deepface --rm -it --name ai-world $(IMAGE_NAME):$(IMAGE_TAG)-$(GIT_COMMIT); \
 	fi
 # Define the base command
 base:

diff --git a/detectron/demo/main.py b/detectron/demo/main.py
@@ -527,4 +527,4 @@ def do_search(selected_dbs,user_input):
 if __name__ == "__main__":
     demo = create_ui()
     demo.queue()
-    demo.launch()
+    demo.launch(server_name="0.0.0.0")