add space

app.py

@@ -4,6 +4,7 @@ import sys
 from typing import Sequence, Mapping, Any, Union
 import torch
 import gradio as gr
+import spaces
 
 from huggingface_hub import hf_hub_download
 from huggingface_hub import snapshot_download
@@ -266,12 +267,9 @@ saveimage = NODE_CLASS_MAPPINGS["SaveImage"]()
 imageupscalewithmodel = NODE_CLASS_MAPPINGS["ImageUpscaleWithModel"]()
 
 
-# def main():
-def generate_image(model_image, hairstyle_template_image):
-
-
-
 
+@spaces.GPU(duration=60)
+def generate_image(model_image, hairstyle_template_image):
     with torch.inference_mode():
         cliptextencode_52 = cliptextencode.encode(
             text="multiple_hands, multiple_legs, multiple_girls\nlow quality, blurry, out of focus, distorted, unrealistic, extra limbs, missing limbs, deformed hands, deformed fingers, extra fingers, long neck, unnatural face, bad anatomy, bad proportions, poorly drawn face, poorly drawn eyes, asymmetrical eyes, extra eyes, extra head, floating objects, watermark, text, logo, jpeg artifacts, overexposed, underexposed, harsh lighting, bad posture, strange angles, unnatural expressions, oversaturated colors, messy hair, unrealistic skin texture, wrinkles inappropriately placed, incorrect shading, pixelation, complex background, busy background, detailed background, crowded scene, clutter, messy elements, unnecessary objects, overlapping objects, intricate patterns, vibrant colors, high contrast, graffiti, shadows, reflections, multiple layers, unrealistic lighting, overexposed areas.",