Update app.py

app.py

@@ -17,7 +17,7 @@ def query(payload):
     response = requests.request("POST", API_URL, headers=headers, data=data)
     try:
         data = json.loads(response.content.decode("utf-8"))
-        response = data[0]['generated_text'].split('<|assistant|> ')[1]
+        response = data[0]['generated_text']
         return response
     except:
         return response