Spaces:
Running
Running
Update app.py
Browse files
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
|
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:
|