Update app.py

app.py

@@ -19,6 +19,7 @@ from base64 import b64encode
 from pathlib import Path
 import openai
 import logging
+import grpc
 
 try:
   import face_recognition
@@ -190,8 +191,10 @@ openai.api_key = os.environ['OPENAI_KEY']
 PROMPT = "colorful portrait 25 year bespectacled woman with long, curly skyblue hair and bright green eyes. She has a small, upturned nose and a freckled complexion. She is approximately 5'5 tall and has a thin build"
 def generate(PROMPT,model):
 #     PROMPT = "An eco-friendly computer from the 90s in the style of vaporwave""Dall-E","StableDiffusion"
-  
-  img = generate_stability(PROMPT)
+  try:
+      img = generate_stability(PROMPT)
+  except grpc._channel._MultiThreadedRendezvous:
+      raise gr.Error("Invalid prompts detected")
   return np.asarray(img)