Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from gradio_client import Client
-import uuid
+import os
 
 MORE = """ ## TRY Other Models
         ### JARVIS: Your VOICE Assistant -> https://huggingface.co/spaces/KingNish/JARVIS
@@ -25,9 +25,9 @@ def generate_image(prompt, base="Realistic", motion="", step=8, progress=gr.Prog
     except ValueError as e:
         # Handle GPU quota exceeded error
         if 'You have exceeded your GPU quota' in str(e):
-            return "Error: GPU quota exceeded. Please retry later."
+            raise gr.Error("GPU Quota exceeded. Please try again later.")
         else:
-            return f"Error: {str(e)}"
+            raise gr.Error(str(e))
 
 # Gradio Interface
 with gr.Blocks(css="style.css") as demo: