Update app.py

app.py

@@ -11,9 +11,8 @@ model = load_model(downloaded_files[0], downloaded_files[1])
     
 def process_cat_embs(style):
     device = torch.device("cpu")
-    style = float(style)
-    embedding = torch.tensor([style, 1 - style], device=device)
-    return embedding
+    cat_embs = torch.tensor([float(c) for c in cat_embs.split(',')]).to(device)
+    return cat_embs
 
 
 @spaces.GPU