Update app.py

app.py

@@ -14,13 +14,15 @@ def app(input, filter_strategy, relevence_threshold, k ):
     
     res = classifier_pipeline(input, **kwargs)
 
+    res = {el['label']:el['score'] for el in res} # convert to expected format for gradio interface
+
     if filter_strategy == 'soglia_di_confidenza':
         res = {k:v for k,v in res.items() if k >= relevence_threshold}
 
     return res
 
 iface = gr.Interface(fn=app, 
-                         inputs=["text", gr.Radio(['soglia_di_confidenza', 'top_k'], label='come vuoi fiiltrare i risultati'), gr.Slider(0, 1, label='soglia di confidenza'), gr.Slider(1, 21, step= 1, label='k')], 
+                         inputs=["text", gr.Radio(['soglia_di_confidenza', 'top_k'], value='soglia_di_confidenza', label='come vuoi fiiltrare i risultati'), gr.Slider(0, 1, value=0.5,label='soglia di confidenza'), gr.Slider(1, 21, step= 1,, value=4, label='k')], 
                      outputs="label")
 
 iface.launch()