Update app.py

app.py

@@ -96,6 +96,7 @@ print("Loading model... This may take a few minutes.")
 generator = SketchToRealisticFace()
 print("Model loaded successfully!")
 
+@spaces.GPU
 def generate_face(sketch_image, custom_prompt, seed):
     """Wrapper function for Gradio interface"""
     try:
@@ -103,11 +104,7 @@ def generate_face(sketch_image, custom_prompt, seed):
         seed_int = int(seed) if seed else None
         
         # Generate the realistic face
-        realistic_face, control_image = generator.generate(
-            sketch_image=sketch_image,
-            custom_prompt=custom_prompt,
-            seed=seed_int
-        )
+        realistic_face, control_image = generator.generate(sketch_image=sketch_image,custom_prompt=custom_prompt,seed=seed_int)
         
         return realistic_face, control_image