Update app.py

app.py

@@ -40,11 +40,12 @@ def generate(
     if formatted_prompt in database:
         response = database[formatted_prompt]
     else:
-        response = next(client.text_generation(formatted_prompt, details=True, return_full_text=False)).token.text
-        database[formatted_prompt] = response
+        response = client.text_generation(formatted_prompt, details=True, return_full_text=False)
+        response_text = response.generated_tokens[0].text
+        database[formatted_prompt] = response_text
         save_database(database)  # Save the updated database
 
-    yield response
+    yield response_text
 
 css = """
   #mkd {