bai4578 commited on
Commit
07dfabc
·
verified ·
1 Parent(s): ce947b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -262,13 +262,13 @@ def generate_text(text, prompt_type):
262
  selected_prompt = prompts.get(prompt_type, "You are a helpful assistant. Process the following text.")
263
 
264
  response = client.chat.completions.create(
265
- model="mixtral-8x7b-32768",
266
  messages=[
267
  {"role": "system", "content": selected_prompt},
268
  {"role": "user", "content": text}
269
  ],
270
- max_tokens=1000,
271
- temperature=0.3
272
  )
273
 
274
  return response.choices[0].message.content
 
262
  selected_prompt = prompts.get(prompt_type, "You are a helpful assistant. Process the following text.")
263
 
264
  response = client.chat.completions.create(
265
+ model="llama-3.3-70b-versatile",
266
  messages=[
267
  {"role": "system", "content": selected_prompt},
268
  {"role": "user", "content": text}
269
  ],
270
+ max_tokens=4000,
271
+ temperature=0.1
272
  )
273
 
274
  return response.choices[0].message.content