Update app.py

app.py

@@ -41,7 +41,8 @@ def inference(input_sentence, max_length, temperature, greedy_decoding, top_k, t
     data = query(
         payload
     )
-    return data[0]['generated_text'][len(input_sentence):]
+    print(data)
+    return data[0]
 
 
 gr.Interface(