import bugfix

defog-ai · Dec 6, 2023 · fb2e918 · fb2e918
1 parent 4a9ec72
commit fb2e918
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/defog/serve.py b/defog/serve.py
@@ -9,7 +9,8 @@
 import os
 import json
 import sqlparse
-import tqdm
+from tqdm import tqdm
+import sys
 
 try:
     from llama_cpp import Llama
@@ -41,7 +42,8 @@
 
     llm = Llama(model_path=filepath, n_gpu_layers=1)
 except Exception as e:
-    print("")
+    print("An error occured when trying to load the model!")
+    sys.exit(1)
 
 
 app = FastAPI()