Munaf1987 commited on
Commit
fdc7c7f
·
verified ·
1 Parent(s): 5ff26f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -41,7 +41,11 @@ except ImportError:
41
  class ProfessionalCartoonFilmGenerator:
42
  def __init__(self):
43
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
44
- self.temp_dir = tempfile.mkdtemp()
 
 
 
 
45
 
46
  # Model configurations for ZeroGPU optimization
47
  self.models_loaded = False
 
41
  class ProfessionalCartoonFilmGenerator:
42
  def __init__(self):
43
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
44
+
45
+ # Use a persistent directory for Hugging Face Spaces
46
+ # Files here can be served by Gradio
47
+ self.temp_dir = "./outputs"
48
+ os.makedirs(self.temp_dir, exist_ok=True)
49
 
50
  # Model configurations for ZeroGPU optimization
51
  self.models_loaded = False