Print Error message

app.py

@@ -23,6 +23,8 @@ def get_completion(prompt, temperature=0.6):
     if output and isinstance(output, list) and "generated_text" in output[0]:
         return output[0]["generated_text"]
     # Debugging
+    if "error" in output:
+        return output["error"]
     return str(output.keys())
 
 def main():