Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -11,6 +11,7 @@ model = load_model(downloaded_files[0], downloaded_files[1])
|
|
11 |
|
12 |
def process_cat_embs(style):
|
13 |
device = torch.device("cpu")
|
|
|
14 |
embedding = torch.tensor([style, 1 - style], device=device)
|
15 |
return embedding
|
16 |
|
|
|
11 |
|
12 |
def process_cat_embs(style):
|
13 |
device = torch.device("cpu")
|
14 |
+
style = float(style)
|
15 |
embedding = torch.tensor([style, 1 - style], device=device)
|
16 |
return embedding
|
17 |
|