Fix import issues in desktop image builds (#343)

khoj-ai · Jul 26, 2023 · 23d77ee · 23d77ee
1 parent 8dcc210
commit 23d77ee
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/Khoj.spec b/Khoj.spec
@@ -19,6 +19,9 @@ datas += copy_metadata('filelock')
 datas += copy_metadata('numpy')
 datas += copy_metadata('tokenizers')
 datas += copy_metadata('pillow')
+datas += copy_metadata('huggingface_hub')
+datas += copy_metadata('safetensors')
+datas += copy_metadata('pyyaml')
 
 block_cipher = None
 

diff --git a/src/khoj/main.py b/src/khoj/main.py
@@ -75,7 +75,8 @@ def run():
         start_server(app, host=args.host, port=args.port, socket=args.socket)
     else:
         from PySide6 import QtWidgets
-        from PySide6.QtCore import QThread, QTimer
+        from PySide6.QtCore import QTimer
+
         from khoj.interface.desktop.main_window import MainWindow, ServerThread
         from khoj.interface.desktop.system_tray import create_system_tray