Spaces:

projectlosangeles
/

Chords-Progressions-Transformer

Sleeping

projectlosangeles commited on Apr 16

Commit

d8c7429

verified ·

1 Parent(s): 040ea86

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -180,7 +180,7 @@ def generate_chords(chords,
     x = torch.LongTensor([718] + cho_prg + [719]).to(DEVICE)
     with ctx:
-        out = prg_model.generate(x,
                                  2048-len(cho_prg)+2,
                                  temperature=input_temperature,
                                  filter_logits_fn=top_p,

     x = torch.LongTensor([718] + cho_prg + [719]).to(DEVICE)
     with ctx:
+        out = tex_model.generate(x,
                                  2048-len(cho_prg)+2,
                                  temperature=input_temperature,
                                  filter_logits_fn=top_p,