Hugging Face's logo

Spaces:
hshetty
/
movie-poster-generator
Runtime error

App Files Community
hshetty commited on
Commit
3c89335
·
1 Parent(s): f488959

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -9,6 +9,8 @@ import os
9
  device = "cuda"
10
  generator = torch.Generator(device=device)
11
 
 
 
12
  HF_TOKEN = os.getenv('HF_TOKEN')
13
  hf_writer =gr.HuggingFaceDatasetSaver(HF_TOKEN, "dst-movie-poster-demo")
14
 
 
9
  device = "cuda"
10
  generator = torch.Generator(device=device)
11
 
12
+ seed = 496012807434005 #generator.seed()
13
+ generator = generator.manual_seed(seed)
14
  HF_TOKEN = os.getenv('HF_TOKEN')
15
  hf_writer =gr.HuggingFaceDatasetSaver(HF_TOKEN, "dst-movie-poster-demo")
16