debisoft commited on
Commit
3136f0e
·
1 Parent(s): 6b1c54c
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ def get_completion(prompt):
36
 
37
  def greet(input):
38
  response = get_completion(input)
39
- gen_text = response[0].generated_text
40
  return gen_text
41
  #return json.dumps(response)
42
 
 
36
 
37
  def greet(input):
38
  response = get_completion(input)
39
+ gen_text = response["predictions"][0]["generated_text"]
40
  return gen_text
41
  #return json.dumps(response)
42