YC-Chen commited on
Commit
90789c4
1 Parent(s): 66f940a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -41,12 +41,10 @@ HEADERS = {
41
  "Content-Type": "application/json",
42
  }
43
 
44
- MODEL_NAME="breeze-7b-instruct-v01"
45
  PRESENCE_PENALTY=0
46
  FREQUENCY_PENALTY=1.1
47
 
48
- model_name = "MediaTek-Research/Breeze-7B-Instruct-v0.1"
49
- tokenizer = AutoTokenizer.from_pretrained(model_name)
50
 
51
  def refusal_condition(query):
52
  # 不要再問這些問題啦!
@@ -142,7 +140,7 @@ with gr.Blocks() as demo:
142
  yield history
143
  else:
144
  data = {
145
- "model": MODEL_NAME,
146
  "prompt": str(message),
147
  "temperature": float(temperature),
148
  "n": 1,
 
41
  "Content-Type": "application/json",
42
  }
43
 
 
44
  PRESENCE_PENALTY=0
45
  FREQUENCY_PENALTY=1.1
46
 
47
+ tokenizer = AutoTokenizer.from_pretrained("MediaTek-Research/Breeze-7B-Instruct-v0_1")
 
48
 
49
  def refusal_condition(query):
50
  # 不要再問這些問題啦!
 
140
  yield history
141
  else:
142
  data = {
143
+ "model": "breeze-7b-instruct-v01",
144
  "prompt": str(message),
145
  "temperature": float(temperature),
146
  "n": 1,