0516_zerogpu

app.py

@@ -173,7 +173,7 @@ def handle_prompts(prompts_array):
     return prompts
     
 
-
+@spaces.GPU
 def generate_image(experiment_name, label, prompts_array, chose_emb):
     prompts = handle_prompts(prompts_array)
     
@@ -275,6 +275,7 @@ def get_example():
     ]
     return case
 
+@spaces.GPU
 def run_for_examples(example_emb, gender_GAN, choice, prompts_array):
     prompts = handle_prompts(prompts_array)
     label, experiment_name = response(choice, gender_GAN)

requirements.txt

@@ -12,4 +12,5 @@ opencv-python
 opencv-contrib-python
 gradio
 huggingface_hub==0.22.2
-IPython
+IPython
+spaces