add space gpy

app.py

@@ -4,7 +4,7 @@ import random
 from PIL import Image
 import os
 
-# import spaces
+import spaces
 from diffusers import StableDiffusion3Pipeline
 import torch
 from peft import PeftModel
@@ -40,7 +40,8 @@ pipe.transformer.load_adapter(lora_models["Text Rendering"], adapter_name="Text
 pipe.transformer.load_adapter(lora_models["Human Prefer"], adapter_name="Human Prefer")
 pipe = pipe.to(device)
 
-COUNTER_FILE = os.path.join(os.getcwd(),"model_call_counter.txt")
+# COUNTER_FILE = os.path.join(os.getcwd(),"model_call_counter.txt")
+COUNTER_FILE = os.path.join("/data/model_call_counter.txt")
 
 def get_call_count():
     if not os.path.exists(COUNTER_FILE):
@@ -90,7 +91,7 @@ def create_grid_image(images):
     
     return grid_image
 
-# @spaces.GPU #[uncomment to use ZeroGPU]
+@spaces.GPU
 def infer(
     prompt,
     seed,

model_call_counter.txt

@@ -1 +0,0 @@
-256