invishalgupta commited on
Commit
74ef659
1 Parent(s): 528026f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -6,7 +6,11 @@ from huggingface_hub import hf_hub_download
6
 
7
  #MODEL SETTINGS also for DISPLAY
8
  convHistory = ''
9
- modelfile = "model/stablelm-zephyr-3b.Q4_K_M.gguf"
 
 
 
 
10
  repetitionpenalty = 1.15
11
  contextlength=4096
12
  logfile = 'StableZephyr3b_logs.txt'
 
6
 
7
  #MODEL SETTINGS also for DISPLAY
8
  convHistory = ''
9
+
10
+ modelfile = hf_hub_download(
11
+ repo_id=os.environ.get("REPO_ID", "TheBloke/stablelm-zephyr-3b-GGUF"),
12
+ filename=os.environ.get("MODEL_FILE", "stablelm-zephyr-3b.Q4_K_M.gguf"),
13
+ )
14
  repetitionpenalty = 1.15
15
  contextlength=4096
16
  logfile = 'StableZephyr3b_logs.txt'