sugiv commited on
Commit
6d7f3ae
·
1 Parent(s): 581c890

Fixing generate and random problem API

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -105,7 +105,7 @@ Here's the complete Python function implementation:
105
 
106
  generated_text = ""
107
  for chunk in llm(full_prompt, stream=True, **generation_kwargs):
108
- generated_text += chunk["choices"]["text"]
109
 
110
  formatted_code = extract_and_format_code(generated_text)
111
  return {"solution": formatted_code}
 
105
 
106
  generated_text = ""
107
  for chunk in llm(full_prompt, stream=True, **generation_kwargs):
108
+ generated_text += chunk["choices"][0]["text"]
109
 
110
  formatted_code = extract_and_format_code(generated_text)
111
  return {"solution": formatted_code}