Fix ZeroGPU supervisor shutdown

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
+import spaces
 
+@spaces.GPU
 def generate_image(prompt, hf_token, progress=gr.Progress(track_tqdm=True)):
     if not hf_token:
         raise gr.Error("Please enter your Hugging Face API Token!")

requirements.txt

@@ -1,2 +1,3 @@
 gradio
-huggingface_hub
+huggingface_hub
+spaces