Update app.py

dgnsfgndghkw4ygafb

app.py

@@ -24,7 +24,7 @@ COMF_PATH = config('COMF_PATH')
 
 import torch
 
-#import spaces
+import spaces
 
 print(f"Is CUDA available: {torch.cuda.is_available()}")
 print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
@@ -84,7 +84,7 @@ def check_server_ready(port):
 
 
 
-#@spaces.GPU(duration=240)
+@spaces.GPU(duration=240)
 def generate_image(prompt, image):
     prefix_filename = str(random.randint(0, 999999))
     prompt = prompt.replace('ComfyUI', prefix_filename)