aledigt commited on
Commit
33b63d2
1 Parent(s): c46fe07

fix: generate params

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -102,7 +102,15 @@ def load_recommender(path, start_page=1):
102
 
103
  def generate_text(prompt):
104
 
105
- data = {"messages":[{"role":"user", "content": prompt}]}
 
 
 
 
 
 
 
 
106
  #print(data)
107
  r = requests.post(api_url, json = data)
108
  completions = r.json()
 
102
 
103
  def generate_text(prompt):
104
 
105
+ data = {
106
+ "frequency_penalty": 0,
107
+ "model": "gpt-3.5-turbo",
108
+ "presence_penalty": 0,
109
+ "temperature":1,
110
+ "top_p": 1,
111
+ "messages":[{"role":"user", "content": prompt}]
112
+ }
113
+
114
  #print(data)
115
  r = requests.post(api_url, json = data)
116
  completions = r.json()