Update app.py

app.py

@@ -35,8 +35,13 @@ def predict(text, seed, out_seq_length, min_gen_length, sampling_strategy,
     }
 
     response = requests.request("POST", url, headers=headers, data=payload)
-
-    return response.json()
+    
+    answer = response.json()['result']['output']['raw']
+    if type(answer) == list:
+        answer = answer[0]
+    
+    answer = answer.replace('</s>', '')
+    return answer
 
 
 if __name__ == "__main__":