Update app.py

app.py

@@ -36,7 +36,7 @@ def predict(text, seed, out_seq_length, min_gen_length, sampling_strategy,
 
     response = requests.request("POST", url, headers=headers, data=payload)
 
-    return response.json()['result']['output']['raw']
+    return response.json()
 
 
 if __name__ == "__main__":