JuyeopDang commited on
Commit
24ff1e4
·
verified ·
1 Parent(s): 0c39b3b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ if __name__ == "__main__":
47
  clip = load_model_from_HF(clip, repo_id, "asian-composite-fine-tuned-koclip.pth", True)
48
  dm = load_model_from_HF(dm, repo_id, "asian-composite-clip-ldm.pth", True)
49
 
50
- def generate_image(y, gamma, dm):
51
  images = dm.sample(2, y = y, gamma = gamma)
52
  images = images.permute(0, 2, 3, 1)
53
  if type(images) is torch.Tensor:
 
47
  clip = load_model_from_HF(clip, repo_id, "asian-composite-fine-tuned-koclip.pth", True)
48
  dm = load_model_from_HF(dm, repo_id, "asian-composite-clip-ldm.pth", True)
49
 
50
+ def generate_image(y, gamma):
51
  images = dm.sample(2, y = y, gamma = gamma)
52
  images = images.permute(0, 2, 3, 1)
53
  if type(images) is torch.Tensor: