Update app.py

app.py

@@ -35,7 +35,8 @@ def respond(
         top_p=top_p,
         stream=True,
     ):
-        token = message['generated_text'].replace(prompt, '')
+        # Since the message is a string, no need for indexing
+        token = message.replace(prompt, '')
         response += token
         yield response