wop commited on
Commit
093ec37
1 Parent(s): 75976ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ def generate(
41
  response = database[formatted_prompt]
42
  else:
43
  response = client.text_generation(formatted_prompt, details=True, return_full_text=False)
44
- response_text = response.generated_tokens[0].text
45
  database[formatted_prompt] = response_text
46
  save_database(database) # Save the updated database
47
 
 
41
  response = database[formatted_prompt]
42
  else:
43
  response = client.text_generation(formatted_prompt, details=True, return_full_text=False)
44
+ response_text = response.generated_text
45
  database[formatted_prompt] = response_text
46
  save_database(database) # Save the updated database
47