SarmaHighOnAI commited on
Commit
676d2b9
·
verified ·
1 Parent(s): 04d8bc5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def generate_text(request: Request):
18
  messages = [{"role": "user", "content": request.prompt}]
19
  response = client.chat_completion(
20
  messages=messages,
21
- model="HuggingFaceTB/SmolLM2-1.7B-Instruct",
22
  max_tokens=500
23
  )
24
  return {"response": response.choices[0].message.content}
 
18
  messages = [{"role": "user", "content": request.prompt}]
19
  response = client.chat_completion(
20
  messages=messages,
21
+ model="meta-llama/Llama-3.1-8B-Instruct",
22
  max_tokens=500
23
  )
24
  return {"response": response.choices[0].message.content}