Update app.py

app.py

@@ -12,15 +12,22 @@ client = Client("KingNish/Instant-Video")
 
 # Function
 def generate_image(prompt, base="Realistic", motion="", step=8, progress=gr.Progress()):
-    result = client.predict(
-        prompt=prompt,
-        base=base,
-        motion=motion,
-        step=step,
-        api_name="/generate_image_1"
-    )
-    video_path = result["video"]
-    return video_path
+    try:
+        result = client.predict(
+            prompt=prompt,
+            base=base,
+            motion=motion,
+            step=step,
+            api_name="/generate_image_1"
+        )
+        video_path = result["video"]
+        return video_path
+    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."
+        else:
+            return f"Error: {str(e)}"
 
 # Gradio Interface
 with gr.Blocks(css="style.css") as demo: