add version choice

app.py

@@ -11,11 +11,19 @@ def get_x(df):
 def get_y(df):
     return df.pattern
    
-learn = load_learner('xresnet50_pitch3_removeSilence.pkl')
+learn_removeSilence = load_learner('xresnet50_pitch3_removeSilence.pkl')
 
-labels = learn.dls.vocab
+learn_plain = load_learner('xresnet50_pitch3.pkl')
 
-def predict(Record, Upload):
+labels = learn_removeSilence.dls.vocab
+
+def process(Record, Upload, version):
+    if version == 'remove silence':
+        return predict(Record, Upload, learn_removeSilence)
+    elif version == 'plain':
+        return predict(Record, Upload, learn_plain)
+
+def predict(Record, Upload, learn):
     if Upload: path = Upload
     else: path = Record
     spec,pred,pred_idx,probs = learn.predict(str(path), with_input=True)
@@ -35,5 +43,10 @@ examples = [['代わる.mp3'],['大丈夫な.mp3'],['熱くない.mp3'], ['あ
 
 enable_queue=True
 
-gr.Interface(fn=predict,inputs=[gr.inputs.Audio(source='microphone', type='filepath', optional=True), gr.inputs.Audio(source='upload', type='filepath', optional=True)], outputs=  [gr.outputs.Label(num_top_classes=3), gr.outputs.Image(type="plot", label='Spectrogram')], title=title,description=description,article=article,examples=examples).launch(debug=True, enable_queue=enable_queue)
+gr.Interface(fn=predict,
+inputs=[gr.inputs.Audio(source='microphone', type='filepath', optional=True),
+gr.inputs.Audio(source='upload', type='filepath', optional=True),
+gr.inputs.Radio(choices=['plain','remove silence'], type="value", default='remove silence', label='version')
+],
+outputs=  [gr.outputs.Label(num_top_classes=3), gr.outputs.Image(type="plot", label='Spectrogram')], title=title,description=description,article=article,examples=examples).launch(debug=True, enable_queue=enable_queue)