Annalyn Ng commited on
Commit
449ebd8
1 Parent(s): a23ebfa

remove result rounding

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -34,7 +34,7 @@ def eval_prob(target_word, text):
34
  # Get probability of target word filling the MASK
35
  result = float(probs[idx])
36
 
37
- return round(result, 5)
38
 
39
  gr.Interface(
40
  fn=eval_prob,
@@ -51,7 +51,7 @@ gr.Interface(
51
  ],
52
  outputs="number",
53
  title="Chinese Sentence Grading",
54
- ).launch()
55
 
56
  # Plot bar chart of probs x target_words to find optimal cutoff
57
 
 
34
  # Get probability of target word filling the MASK
35
  result = float(probs[idx])
36
 
37
+ return result
38
 
39
  gr.Interface(
40
  fn=eval_prob,
 
51
  ],
52
  outputs="number",
53
  title="Chinese Sentence Grading",
54
+ ).launch(share=True)
55
 
56
  # Plot bar chart of probs x target_words to find optimal cutoff
57