update

.gitignore

@@ -0,0 +1,2 @@
+__pycache__
+weights/

app.py

@@ -10,7 +10,7 @@ from sixdrepnet.model import SixDRepNet
 import pixel_generator.vec2face.model_vec2face as model_vec2face
 import torch
 import os
-# import spaces
+import spaces
 from time import time
 
 
@@ -81,7 +81,7 @@ def initialize_models():
 
     return generator, id_model, pose_model, quality_model
 
-# @spaces.GPU
+@spaces.GPU
 def image_generation(input_image, quality, random_perturbation, sigma, dimension, progress=gr.Progress()):
     generator, id_model, pose_model, quality_model = initialize_models()
 
@@ -121,7 +121,7 @@ def image_generation(input_image, quality, random_perturbation, sigma, dimension
 
     return generated_images
 
-# @spaces.GPU
+@spaces.GPU
 def process_input(image_input, num1, num2, num3, num4, random_seed, target_quality, random_perturbation, sigma, progress=gr.Progress()):
     # Ensure all dimension numbers are within [0, 512)
     num1, num2, num3, num4 = [max(0, min(int(n), 511)) for n in [num1, num2, num3, num4]]
@@ -294,7 +294,7 @@ def main():
         """
         gr.Markdown(article)
 
-    demo.launch(share=True)
+    demo.launch()
 
 
 if __name__ == "__main__":