arampacha commited on
Commit
f3d44ff
·
1 Parent(s): c6b7aba

add options

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -30,10 +30,14 @@ parameters = {
30
  "min_length":None,
31
  "max_length":100,
32
  "top_p":0.92,
 
33
  "repetition_penalty":None,
34
  }
35
 
36
- # options = {}
 
 
 
37
 
38
  def on_input():
39
  # st.write("Input changed")
@@ -50,6 +54,7 @@ def on_input():
50
  "generated_responses": st.session_state.generated_responses,
51
  },
52
  "parameters": parameters,
 
53
  }
54
  # result = fake_query(payload)
55
  result = query(payload)
 
30
  "min_length":None,
31
  "max_length":100,
32
  "top_p":0.92,
33
+ "temperature":1.0,
34
  "repetition_penalty":None,
35
  }
36
 
37
+ options = {
38
+ "use_cache":False,
39
+ "wait_for_model":False
40
+ }
41
 
42
  def on_input():
43
  # st.write("Input changed")
 
54
  "generated_responses": st.session_state.generated_responses,
55
  },
56
  "parameters": parameters,
57
+ "options":options,
58
  }
59
  # result = fake_query(payload)
60
  result = query(payload)