Upload app.py

app.py

@@ -125,6 +125,11 @@ class ABCTokenizer():
 
 def generate_abc(control_codes, prefix, num_tunes, max_length, top_p, temperature, seed):
 
+    try:
+        seed = int(seed)
+    except:
+        seed = None
+
     prefix = unidecode(control_codes + prefix)
     tokenizer = ABCTokenizer()
     model = GPT2LMHeadModel.from_pretrained('sander-wood/tunesformer').to(device)
@@ -175,7 +180,7 @@ input_num_tunes = gr.inputs.Slider(minimum=1, maximum=10, step=1, default=3, lab
 input_max_length = gr.inputs.Slider(minimum=10, maximum=1000, step=10, default=500, label="Max Length")
 input_top_p = gr.inputs.Slider(minimum=0.0, maximum=1.0, step=0.05, default=0.9, label="Top P")
 input_temperature = gr.inputs.Slider(minimum=0.0, maximum=1.0, step=0.05, default=0.9, label="Temperature")
-input_seed = gr.inputs.Textbox(lines=1, label="Seed", default="None")
+input_seed = gr.inputs.Textbox(lines=1, label="Seed (int)", default="None")
 output_abc = gr.outputs.Textbox(label="Generated Tunes")
 
 gr.Interface(generate_abc,