nimatov commited on
Commit
dcdcb87
·
verified ·
1 Parent(s): e986507

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ def load_beat_model():
23
  beat_service = _Beat_tracking_service()
24
  return beat_service
25
 
26
- @st.cache_resource
27
  def load_image_model():
28
  torch.cuda.empty_cache()
29
  pipeline = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, revision="fp16")
 
23
  beat_service = _Beat_tracking_service()
24
  return beat_service
25
 
26
+ # @st.cache_resource
27
  def load_image_model():
28
  torch.cuda.empty_cache()
29
  pipeline = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, revision="fp16")