sguertl commited on
Commit
52de3c1
·
verified ·
1 Parent(s): 1dd10ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,7 +11,7 @@ from huggingface_hub import InferenceClient
11
  HF_MODEL_MISTRAL = "mistralai/Mistral-7B-Instruct-v0.3"
12
  HF_MODEL_LLAMA = "meta-llama/Llama-3.3-70B-Instruct"
13
 
14
- PROMPTS_DOC_URL = "https://docs.google.com/document/d/17rB_0BGQ4DGT7pwOV8O58sNvzBWgywq6ZDgzyJ9pmjs/export?format=txt"
15
 
16
  STEP1_SYSTEM_PROMPT = "STEP1 SYSPROMPT"
17
  STEP1_USER_PROMPT = "STEP1 USERPROMPT"
@@ -24,7 +24,7 @@ STEP3B_USER_PROMPT = "STEP3B USERPROMPT"
24
 
25
 
26
  def fetch_prompts_from_google_doc():
27
- response = requests.get(PROMPTS_DOC_URL)
28
  if response.status_code != 200:
29
  raise Exception("Failed to fetch document")
30
 
 
11
  HF_MODEL_MISTRAL = "mistralai/Mistral-7B-Instruct-v0.3"
12
  HF_MODEL_LLAMA = "meta-llama/Llama-3.3-70B-Instruct"
13
 
14
+ UML_PROMPTS_DOC_URL = os.environ['UML_PROMPTS_DOC_URL']
15
 
16
  STEP1_SYSTEM_PROMPT = "STEP1 SYSPROMPT"
17
  STEP1_USER_PROMPT = "STEP1 USERPROMPT"
 
24
 
25
 
26
  def fetch_prompts_from_google_doc():
27
+ response = requests.get(UML_PROMPTS_DOC_URL)
28
  if response.status_code != 200:
29
  raise Exception("Failed to fetch document")
30