Nymbo commited on
Commit
b32bd6a
·
verified ·
1 Parent(s): f804af3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ import json
5
  import torch
6
  from diffusers import FluxPipeline, AutoencoderKL
7
  from live_preview_helpers import flux_pipe_call_that_returns_an_iterable_of_images
8
- # import spaces
9
 
10
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
11
 
@@ -55,7 +55,7 @@ def get_random_world_description():
55
  def get_random_persona_description():
56
  return ds.shuffle().select([100])[0]["persona"]
57
 
58
- # @spaces.GPU(duration=75)
59
  def infer_flux(character_json):
60
  for image in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
61
  prompt=character_json["appearance"],
 
5
  import torch
6
  from diffusers import FluxPipeline, AutoencoderKL
7
  from live_preview_helpers import flux_pipe_call_that_returns_an_iterable_of_images
8
+ import spaces
9
 
10
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
11
 
 
55
  def get_random_persona_description():
56
  return ds.shuffle().select([100])[0]["persona"]
57
 
58
+ @spaces.GPU(duration=75)
59
  def infer_flux(character_json):
60
  for image in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
61
  prompt=character_json["appearance"],