nroggendorff commited on
Commit
cf81fc2
1 Parent(s): 7613e30

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,6 +3,7 @@ from transformers import pipeline
3
  import torch
4
 
5
  import re
 
6
 
7
  import spaces
8
 
@@ -11,8 +12,7 @@ torch.set_default_device("cuda")
11
  model_id = "glides/llama-eap"
12
  pipe = pipeline("text-generation", model=model_id, device_map="auto")
13
 
14
- with open("sys", "r") as f:
15
- system_prompt = f.read()
16
 
17
  def follows_rules(s):
18
  pattern = r'<thinking>.+?</thinking><output>.+?</output><reflecting>.+?</reflecting><refined>.+?</refined>'
 
3
  import torch
4
 
5
  import re
6
+ import os
7
 
8
  import spaces
9
 
 
12
  model_id = "glides/llama-eap"
13
  pipe = pipeline("text-generation", model=model_id, device_map="auto")
14
 
15
+ system_prompt = os.environ["sys"]
 
16
 
17
  def follows_rules(s):
18
  pattern = r'<thinking>.+?</thinking><output>.+?</output><reflecting>.+?</reflecting><refined>.+?</refined>'