jsvs commited on
Commit
9b6493b
·
1 Parent(s): c67003d

update method

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,4 +34,4 @@ def generate(text: str,n: int):
34
  output = pipe("contexto: "+text,num_return_sequences=n,num_beams=n)
35
 
36
  # Return the generated text in a JSON response
37
- return {"output": [output[i]["generated_text"] for i in len(output)]}
 
34
  output = pipe("contexto: "+text,num_return_sequences=n,num_beams=n)
35
 
36
  # Return the generated text in a JSON response
37
+ return {"output": [output[i]["generated_text"] for i in range(len(output))]}