vkovacs commited on
Commit
20f3f13
·
1 Parent(s): fb347fa

testing fixed column widths

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -72,7 +72,7 @@ def predict_wrapper(text, language):
72
  label, probability = predict(sentence, model_id, tokenizer_id)
73
  results.append([sentence, label, probability])
74
 
75
- output_info = f'Prediction made using the <a href="https://huggingface.co/{model_id}">{model_id}</a> model.'
76
  return results, output_info
77
 
78
  with gr.Blocks() as demo:
@@ -83,7 +83,10 @@ with gr.Blocks() as demo:
83
  predict_button = gr.Button("Submit")
84
 
85
  with gr.Column():
86
- result_table = gr.Dataframe(headers=["Sentence", "Prediction", "Confidence"])
 
 
 
87
  model_info = gr.Markdown()
88
 
89
  predict_button.click(
 
72
  label, probability = predict(sentence, model_id, tokenizer_id)
73
  results.append([sentence, label, probability])
74
 
75
+ output_info = f'Prediction was made using the <a href="https://huggingface.co/{model_id}">{model_id}</a> model.'
76
  return results, output_info
77
 
78
  with gr.Blocks() as demo:
 
83
  predict_button = gr.Button("Submit")
84
 
85
  with gr.Column():
86
+ result_table = gr.Dataframe(
87
+ headers=["Sentence", "Prediction", "Confidence"],
88
+ column_widths=["50%", "35%", "15%"]
89
+ )
90
  model_info = gr.Markdown()
91
 
92
  predict_button.click(