Update app.py

app.py

@@ -10,8 +10,8 @@ downloaded_files = [hf_hub_download(repo_id=REPO_ID, filename=f) for f in files]
 model = load_model(downloaded_files[0], downloaded_files[1])
     
 def process_cat_embs(style):
-    device = torch.device("cpu")
-    cat_embs = torch.tensor([float(c) for c in cat_embs.split(',')]).to(device)
+    device = torch.device("gpu")
+    cat_embs = torch.tensor([float(c) for c in style.split(',')]).to(device)
     return cat_embs