update error message

app.py

@@ -187,6 +187,9 @@ def docgen_func(function_code, min_length, max_length, top_k, top_p, temp, repet
     if type(output) is list:
         return f'"""\n{" || ".join(output[0]["generated_text"].split("|"))}\n"""'
     else:
+        msg = str(output)
+        if msg == "{'error': 'Model stmnk/codet5-small-code-summarization-python is currently loading', 'estimated_time': 20}":
+            return msg + '\n \n Please wait for the model to load and try again'
         return str(output)
 
 iface = gr.Interface(