yasserrmd commited on
Commit
a6beba6
·
verified ·
1 Parent(s): e5c80b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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