dexter2389 commited on
Commit
64b34d9
·
1 Parent(s): a00a37a

Access value instead of enum

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ def chat(payload: ChatRequest, request: Request):
61
  ad_fetch_response = client.fetch_ad_units(fetch_payload)
62
 
63
  ai_response = (
64
- payload.model(
65
  [{"role": "user", "content": f"{payload.message}"}], do_sample=False
66
  )[0]
67
  .get("generated_text", [{}, {}])[1]
 
61
  ad_fetch_response = client.fetch_ad_units(fetch_payload)
62
 
63
  ai_response = (
64
+ payload.model.value(
65
  [{"role": "user", "content": f"{payload.message}"}], do_sample=False
66
  )[0]
67
  .get("generated_text", [{}, {}])[1]