bloom

app.py

@@ -8,11 +8,11 @@ def reply(message, history):
     txt = Translation(message, "en")
     if txt.original == "en":
         response = pipe(message)
-        return response[0]["generated_text"].split("Answer\n")[1]
+        return response[0]["generated_text"]
     else:
         translation = txt.translatef()
         response = pipe(translation)
-        t = Translation(response[0]["generated_text"].split("Answer\n")[1], txt.original)
+        t = Translation(response[0]["generated_text"], txt.original)
         res = t.translatef()
         return res