Explicited the error logs for the users

app.py

@@ -20,6 +20,7 @@ def query(payload):
     print(response)
     return json.loads(response.content.decode("utf-8"))
 
+
 def inference(input_sentence, max_length, sample_or_greedy, seed=42):
     if sample_or_greedy == "Sample":
         parameters = {
@@ -43,11 +44,10 @@ def inference(input_sentence, max_length, sample_or_greedy, seed=42):
 
     payload = {"inputs": input_sentence, "parameters": parameters}
 
-    #if API_URL is not None:
     data = query(payload)
-    #else:  # To remove when plugged in a real API
-    #    input_sentence, generation = payload["inputs"].split("\n", 1)
-    #    data = [{"generated_text": input_sentence + generation}]
+
+    if "error" in data:
+        return (None, None, f"<span style='color:red'>ERROR: {data['error']} </span>")
 
     generation = data[0]["generated_text"].split(input_sentence, 1)[1]
     return (
@@ -57,6 +57,7 @@ def inference(input_sentence, max_length, sample_or_greedy, seed=42):
         + generation
         + after_generation,
         data[0]["generated_text"],
+        "",
     )
 
 
@@ -86,9 +87,9 @@ if __name__ == "__main__":
                     submit = gr.Button("Submit")
                     load_image = gr.Button("Generate Image")
             with gr.Column():
-                display_out = gr.HTML(
-                    label="Image"
-                )  # , visible=False) #interactive=True,
+                text_error = gr.Markdown(label="Log information")
+                text_out = gr.Textbox(label="Output")
+                display_out = gr.HTML(label="Image")
                 display_out.set_event_trigger(
                     "load",
                     fn=None,
@@ -97,14 +98,13 @@ if __name__ == "__main__":
                     no_target=True,
                     js=js_load_script,
                 )
-                text_out = gr.Textbox(label="Output")
         with gr.Row():
-           gr.Examples(examples=examples, inputs=[text, tokens, sampling, sampling2])
+            gr.Examples(examples=examples, inputs=[text, tokens, sampling, sampling2])
 
         submit.click(
             inference,
             inputs=[text, tokens, sampling, sampling2],
-            outputs=[display_out, text_out],
+            outputs=[display_out, text_out, text_error],
         )
 
         load_image.click(fn=None, inputs=None, outputs=None, _js=js_save)