eddddyy commited on
Commit
51354f3
·
verified ·
1 Parent(s): 5c1e51e

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +33 -2
config.py CHANGED
@@ -1,7 +1,38 @@
1
  import os
2
 
 
3
  HF_TOKEN = os.getenv("HF_TOKEN")
4
- MODEL_ID = "Qwen/Qwen1.5-1.8B-Chat"
5
-
6
  if not HF_TOKEN:
7
  raise ValueError("❌ HF_TOKEN not set in environment variables.")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import os
2
 
3
+ # Hugging Face Token
4
  HF_TOKEN = os.getenv("HF_TOKEN")
 
 
5
  if not HF_TOKEN:
6
  raise ValueError("❌ HF_TOKEN not set in environment variables.")
7
+
8
+ # Model to Load
9
+ MODEL_ID = os.getenv("MODEL_ID", "Qwen/Qwen1.5-1.8B-Chat") # Allows override via env
10
+
11
+ # Paths and Settings
12
+ TEMP_DIR = os.getenv("TEMP_DIR", "/tmp") # For screenshots, temporary files, etc.
13
+ ENABLE_EXECUTION = os.getenv("ENABLE_EXECUTION", "true").lower() == "true" # toggle command execution
14
+
15
+ # OS Commands (override or customize per OS)
16
+ OS_COMMANDS = {
17
+ "open": {
18
+ "chrome": "google-chrome",
19
+ "firefox": "firefox",
20
+ "notepad": "notepad",
21
+ "calculator": "gnome-calculator",
22
+ "terminal": "gnome-terminal",
23
+ "vscode": "code"
24
+ },
25
+ "close": {
26
+ "chrome": "pkill chrome",
27
+ "firefox": "pkill firefox",
28
+ "notepad": "pkill notepad",
29
+ "calculator": "pkill gnome-calculator",
30
+ "vscode": "pkill code"
31
+ },
32
+ "screenshot": {
33
+ "default": "import -window root screenshot.png"
34
+ },
35
+ "shutdown": "shutdown now",
36
+ "restart": "reboot",
37
+ "lock": "gnome-screensaver-command -l"
38
+ }