Update app.py

app.py

@@ -59,8 +59,18 @@ def generate(query: str, temperature=0.9, max_new_tokens=256, top_p=0.95, repeti
             previous_token = response.token.text
             yield output
         return output
-    except errors.UnknownError:
-        return "Please wait for a while, The OctoCoder model is currently loading... 🐙"
+    except errors.UnknownError as e:
+        print(f"Error: {e}")
+        message = "Please wait for a while, The OctoCoder model is currently loading... 🐙"
+        output = ""
+        for item in message.split(" "):
+            if item == "🐙":
+                output += "🐙"
+                return output
+            else:
+                output += f"{item} "
+            yield output
+        return output
 
 
 def process_example(**krwags):