ARCQUB commited on
Commit
4c39f47
·
verified ·
1 Parent(s): 1af8f14

Update models/pixtral.py

Browse files
Files changed (1) hide show
  1. models/pixtral.py +3 -2
models/pixtral.py CHANGED
@@ -18,8 +18,9 @@ sampling_params = SamplingParams(max_tokens=5000)
18
 
19
 
20
  def load_prompt():
21
- with open("prompts/prompt.txt", "r", encoding="utf-8") as f:
22
- return f.read()
 
23
 
24
 
25
  def try_extract_json(text):
 
18
 
19
 
20
  def load_prompt():
21
+ #with open("prompts/prompt.txt", "r", encoding="utf-8") as f:
22
+ # return f.read()
23
+ return os.getenv("PROMPT_TEXT", "⚠️ PROMPT_TEXT not found in secrets.")
24
 
25
 
26
  def try_extract_json(text):